Revert "CUDA incompatible"
Browse filesThis reverts commit 3dda4e1c6f690d2e7882ecb3a0850429a2624263.
- requirements_docker.txt +3 -3
requirements_docker.txt
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
torch==2.1.1
|
2 |
-
torchvision==0.
|
3 |
-
torchaudio==2.
|
4 |
-
torchtext==0.
|
5 |
torchdata==0.6.1
|
|
|
1 |
torch==2.1.1
|
2 |
+
torchvision==0.15.2
|
3 |
+
torchaudio==2.0.2
|
4 |
+
torchtext==0.15.2
|
5 |
torchdata==0.6.1
|