wenxiang guo commited on
Commit
371afed
·
verified ·
1 Parent(s): 12b6a07

Update pre-requirements.txt

Browse files
Files changed (1) hide show
  1. pre-requirements.txt +4 -0
pre-requirements.txt CHANGED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ --index-url https://download.pytorch.org/whl/cu118
2
+ torch==2.1.0
3
+ torchvision==0.16.0
4
+ torchaudio==2.1.0