Taejun Kim commited on
Commit
b6cafed
1 Parent(s): 1aa0f5b

Use torch with cuda 11.8

Browse files
Files changed (1) hide show
  1. pre-requirements.txt +1 -0
pre-requirements.txt CHANGED
@@ -1,3 +1,4 @@
1
  # --extra-index-url https://download.pytorch.org/whl/cpu
 
2
  torch==2.0.1
3
  torchaudio
 
1
  # --extra-index-url https://download.pytorch.org/whl/cpu
2
+ --extra-index-url https://download.pytorch.org/whl/cu118
3
  torch==2.0.1
4
  torchaudio