shgao commited on
Commit
094f556
1 Parent(s): 74744bd
Files changed (1) hide show
  1. requirements.txt +4 -3
requirements.txt CHANGED
@@ -1,6 +1,7 @@
1
- https://download.pytorch.org/whl/cu111/torch-1.10.1%2Bcu111-cp38-cp38-linux_x86_64.whl
2
- https://download.pytorch.org/whl/cu111/torchvision-0.11.2%2Bcu111-cp38-cp38-linux_x86_64.whl
3
- https://download.pytorch.org/whl/cu111/torchaudio-0.10.1%2Bcu111-cp38-cp38-linux_x86_64.whl
 
4
  numpy==1.23.1
5
  gradio==3.16.2
6
  albumentations==1.3.0
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu117
2
+ torch==1.13.1+cu117
3
+ torchvision==0.14.1+cu117
4
+ torchaudio==0.13.1
5
  numpy==1.23.1
6
  gradio==3.16.2
7
  albumentations==1.3.0