Sudipta Nayak commited on
Commit
6772b36
·
1 Parent(s): 42c0f88
Files changed (1) hide show
  1. requirements_gpu.txt +5 -2
requirements_gpu.txt CHANGED
@@ -11,5 +11,8 @@
11
 
12
 
13
  # CPU-only
14
- torch==torch-1.11.0-cp310-cp310-linux_x86_64.whl
15
- torchvision==torchvision-0.12.0-cp310-cp310-linux_x86_64.whl
 
 
 
 
11
 
12
 
13
  # CPU-only
14
+ # torch==torch-1.11.0-cp310-cp310-linux_x86_64.whl
15
+ # torchvision==torchvision-0.12.0-cp310-cp310-linux_x86_64.whl
16
+
17
+ torch==torch-1.11.0+cu113-cp310-cp310-linux_x86_64.whl
18
+ torchvision==torchvision-0.12.0+cu113-cp310-cp310-linux_x86_64.whl