all-in-one / pre-requirements.txt
Taejun Kim
Use torch with cuda 11.8
b6cafed
raw
history blame
138 Bytes
# --extra-index-url https://download.pytorch.org/whl/cpu
--extra-index-url https://download.pytorch.org/whl/cu118
torch==2.0.1
torchaudio