ylop commited on
Commit
cc1f3e1
·
verified ·
1 Parent(s): 28dbd6d

Fix: Use CUDA 11.8 for better Docker compatibility

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -1,5 +1,5 @@
1
  <<<<<<< HEAD
2
- --extra-index-url https://download.pytorch.org/whl/cu121
3
  =======
4
  --extra-index-url https://download.pytorch.org/whl/cu113
5
  >>>>>>> 7245997 (Force deploy 8M steps)
 
1
  <<<<<<< HEAD
2
+ --extra-index-url https://download.pytorch.org/whl/cu118
3
  =======
4
  --extra-index-url https://download.pytorch.org/whl/cu113
5
  >>>>>>> 7245997 (Force deploy 8M steps)