Revert "CUDA incompatible"
Browse filesThis reverts commit ffdd025d9b3d08fdf21fc183b769b55f8e0d9cff.
- requirements_docker.txt +1 -1
requirements_docker.txt
CHANGED
@@ -1,4 +1,4 @@
|
|
1 |
-
torch==2.
|
2 |
torchvision==0.15.2
|
3 |
torchaudio==2.0.2
|
4 |
torchtext==0.15.2
|
|
|
1 |
+
torch==2.0.1
|
2 |
torchvision==0.15.2
|
3 |
torchaudio==2.0.2
|
4 |
torchtext==0.15.2
|