Update PyTorch extra-index-url to CUDA 11.8
parent
e1277af2ba
commit
a392ec9b2c
|
@ -1,4 +1,4 @@
|
|||
--extra-index-url https://download.pytorch.org/whl/cu117
|
||||
--extra-index-url https://download.pytorch.org/whl/cu118
|
||||
torch==2.0.0
|
||||
torchvision==0.15.0
|
||||
omegaconf
|
||||
|
|
Loading…
Reference in New Issue