haodongli commited on
Commit
ce29703
·
verified ·
1 Parent(s): 0170e3c

Update pre-requirements.txt

Browse files
Files changed (1) hide show
  1. pre-requirements.txt +1 -0
pre-requirements.txt CHANGED
@@ -1,4 +1,5 @@
1
  pip==23.3.1
 
2
  torch==2.0.0 --index-url https://download.pytorch.org/whl/cu118
3
  torchvision==0.15.1 --index-url https://download.pytorch.org/whl/cu118
4
  torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118
 
1
  pip==23.3.1
2
+ pycuda==11.7
3
  torch==2.0.0 --index-url https://download.pytorch.org/whl/cu118
4
  torchvision==0.15.1 --index-url https://download.pytorch.org/whl/cu118
5
  torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118