From 4f8d2a951a2661775e25a47e3dceb596fa2a30a1 Mon Sep 17 00:00:00 2001 From: Shan E Ahmed Raza <13048456+shaneahmed@users.noreply.github.com> Date: Thu, 30 Nov 2023 14:27:26 +0000 Subject: [PATCH] :pushpin: Update dependency link for pytorch cuda version --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 4d5e4b78a..f72cac62f 100644 --- a/setup.py +++ b/setup.py @@ -21,7 +21,7 @@ dependency_links = [] if sys.platform != "darwin": - dependency_links = ["https://download.pytorch.org/whl/cu113"] + dependency_links = ["https://download.pytorch.org/whl/cu118"] setup_requirements = [ "pytest-runner",