Update PyTorch extra-index-url to CUDA 11.8

pull/434/head
eightsuzuki 2024-06-27 17:32:25 +09:00
parent e1277af2ba
commit a392ec9b2c
1 changed files with 1 additions and 1 deletions

View File

@ -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