Sudipta Nayak
commited on
Commit
·
6772b36
1
Parent(s):
42c0f88
changes-1
Browse files- 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
|