feng2022 commited on
Commit
b236172
1 Parent(s): 7caab56

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -4
requirements.txt CHANGED
@@ -23,7 +23,7 @@ scipy
23
  matplotlib
24
  pprintpp
25
 
26
- -f https://download.pytorch.org/whl/cu116
27
- torch
28
- torchvision
29
- torchaudio
 
23
  matplotlib
24
  pprintpp
25
 
26
+ pytorch==1.10.0
27
+ torchvision==0.11.0
28
+ torchaudio==0.10.0
29
+ cudatoolkit=11.3