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

Revert "CUDA incompatible"

Browse files

This reverts commit ffdd025d9b3d08fdf21fc183b769b55f8e0d9cff.

Files changed (1) hide show
  1. requirements_docker.txt +1 -1
requirements_docker.txt CHANGED
@@ -1,4 +1,4 @@
1
- torch==2.1.1
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