From 0e5c3a641a1e3842ca0f7b5868adbc4651666e14 Mon Sep 17 00:00:00 2001 From: Millun Atluri Date: Tue, 14 Nov 2023 11:05:07 +1100 Subject: [PATCH] Revert torch to use cu121 --- installer/lib/installer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/installer/lib/installer.py b/installer/lib/installer.py index 8938a39b3a..0d30756fe8 100644 --- a/installer/lib/installer.py +++ b/installer/lib/installer.py @@ -460,10 +460,10 @@ def get_torch_source() -> (Union[str, None], str): url = "https://download.pytorch.org/whl/cpu" if device == "cuda": - url = "https://download.pytorch.org/whl/cu118" + url = "https://download.pytorch.org/whl/cu121" optional_modules = "[xformers,onnx-cuda]" if device == "cuda_and_dml": - url = "https://download.pytorch.org/whl/cu118" + url = "https://download.pytorch.org/whl/cu121" optional_modules = "[xformers,onnx-directml]" # in all other cases, Torch wheels should be coming from PyPi as of Torch 1.13