Sudipta Nayak
commited on
Commit
·
2da0893
1
Parent(s):
e6702de
torch with CPU support only
Browse files- requirements_gpu.txt +6 -2
requirements_gpu.txt
CHANGED
|
@@ -2,5 +2,9 @@
|
|
| 2 |
-i https://download.pytorch.org/whl/cu113
|
| 3 |
|
| 4 |
# CUDA 11.3
|
| 5 |
-
torch==1.11.0+cu113
|
| 6 |
-
torchvision==0.12.0+cu113
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 2 |
-i https://download.pytorch.org/whl/cu113
|
| 3 |
|
| 4 |
# CUDA 11.3
|
| 5 |
+
# torch==1.11.0+cu113
|
| 6 |
+
# torchvision==0.12.0+cu113
|
| 7 |
+
|
| 8 |
+
#CPU-only version of PyTorch
|
| 9 |
+
torch==1.11.0
|
| 10 |
+
torchvision==0.12.0
|