diff --git a/setup.py b/setup.py index 84c536e08..09464551e 100644 --- a/setup.py +++ b/setup.py @@ -33,7 +33,7 @@ amd_torch_index = "https://download.pytorch.org/whl/rocm5.4.2" The package index to torch built with CUDA. Observe the CUDA version is in this URL. """ -nvidia_torch_index = "https://download.pytorch.org/whl/nightly/cu118" +nvidia_torch_index = "https://download.pytorch.org/whl/cu118" """ The package index to torch built against CPU features.