From 5af026f1f6c56f92c37e1c022b9e942f513c3173 Mon Sep 17 00:00:00 2001 From: Benjamin Berman Date: Thu, 3 Aug 2023 08:23:17 -0700 Subject: [PATCH] use the stable index for nvidia for now since nightlies don't resolve correctly --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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.