JinhuaL1ANG commited on
Commit
d4e5e45
·
verified ·
1 Parent(s): 7773d55

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -4
requirements.txt CHANGED
@@ -13,7 +13,7 @@ librosa
13
  opencv-python
14
  progressbar
15
  # wandb
16
- --extra-index-url https://download.pytorch.org/whl/cu118
17
- torch==2.0.1
18
- torchaudio==2.0.2
19
- torchvision==0.15.2
 
13
  opencv-python
14
  progressbar
15
  # wandb
16
+ --extra-index-url https://download.pytorch.org/whl/cu121
17
+ torch==2.1.0
18
+ torchaudio==2.1.0
19
+ torchvision==0.16.0