JinhuaL1ANG commited on
Commit
502eb4a
·
verified ·
1 Parent(s): c7710a2

Update requirements.txt

Browse files

compatible with sm90 arch

Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -13,7 +13,7 @@ librosa
13
  opencv-python
14
  progressbar
15
  # wandb
16
- --extra-index-url https://download.pytorch.org/whl/cu117
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/cu118
17
  torch==2.0.1
18
  torchaudio==2.0.2
19
  torchvision==0.15.2