Skip to content

Commit cbc905a

Browse files
authored
Update installer.py to use cu118 instead of 121
1 parent 8702a63 commit cbc905a

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

installer/lib/installer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -460,10 +460,10 @@ def get_torch_source() -> (Union[str, None], str):
460460
url = "https://download.pytorch.org/whl/cpu"
461461

462462
if device == "cuda":
463-
url = "https://download.pytorch.org/whl/cu121"
463+
url = "https://download.pytorch.org/whl/cu118"
464464
optional_modules = "[xformers,onnx-cuda]"
465465
if device == "cuda_and_dml":
466-
url = "https://download.pytorch.org/whl/cu121"
466+
url = "https://download.pytorch.org/whl/cu118"
467467
optional_modules = "[xformers,onnx-directml]"
468468

469469
# in all other cases, Torch wheels should be coming from PyPi as of Torch 1.13

0 commit comments

Comments
 (0)