tharms commited on
Commit
83afa62
·
1 Parent(s): c73cb30

Revert "CUDA incompatible"

Browse files

This reverts commit 3dda4e1c6f690d2e7882ecb3a0850429a2624263.

Files changed (1) hide show
  1. requirements_docker.txt +3 -3
requirements_docker.txt CHANGED
@@ -1,5 +1,5 @@
1
  torch==2.1.1
2
- torchvision==0.16.1
3
- torchaudio==2.1.1
4
- torchtext==0.16.1
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