feng2022 commited on
Commit
aa011c0
1 Parent(s): 1a7d096

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -24,6 +24,6 @@ matplotlib
24
  pprintpp
25
 
26
  -f https://download.pytorch.org/whl/cu116
27
- torch
28
  torchvision
29
  torchaudio
 
24
  pprintpp
25
 
26
  -f https://download.pytorch.org/whl/cu116
27
+ torch_cuda80
28
  torchvision
29
  torchaudio