MingGatsby commited on
Commit
6685d8d
1 Parent(s): e896750

Update requirement.txt

Browse files
Files changed (1) hide show
  1. requirement.txt +3 -3
requirement.txt CHANGED
@@ -1,6 +1,6 @@
1
  streamlit
2
  gradio
3
  monai
4
- torch==2.0.1 --index-url https://download.pytorch.org/whl/cu117
5
- torchvision==0.15.2 --index-url https://download.pytorch.org/whl/cu117
6
- torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu117
 
1
  streamlit
2
  gradio
3
  monai
4
+ torch==2.0.1 --index-url https://download.pytorch.org/whl/cu118
5
+ torchvision==0.15.2 --index-url https://download.pytorch.org/whl/cu118
6
+ torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu118