From 06313b210a6b9335dcf9e75b8ac9b4155dc6e19c Mon Sep 17 00:00:00 2001 From: Rasmus Oersoe Date: Mon, 9 Oct 2023 13:15:17 +0200 Subject: [PATCH 1/2] adjust gpu versions in requirements.txt --- requirements/torch_gpu.txt | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/requirements/torch_gpu.txt b/requirements/torch_gpu.txt index 03e2a6906..cc9f55588 100644 --- a/requirements/torch_gpu.txt +++ b/requirements/torch_gpu.txt @@ -1,4 +1,4 @@ # Contains packages recommended for functional performance --find-links https://download.pytorch.org/whl/torch_stable.html -torch==2.1.1+cu117 ---find-links https://data.pyg.org/whl/torch-2.1.0+cu117.html +torch==2.1.0+cu118 +--find-links https://data.pyg.org/whl/torch-2.1.0+cu118.html From 9e35a549cc1258b40c5ea6792c61db8d6be05e0d Mon Sep 17 00:00:00 2001 From: Rasmus Oersoe Date: Mon, 9 Oct 2023 13:17:11 +0200 Subject: [PATCH 2/2] update comment --- requirements/torch_gpu.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements/torch_gpu.txt b/requirements/torch_gpu.txt index cc9f55588..1f1abba3f 100644 --- a/requirements/torch_gpu.txt +++ b/requirements/torch_gpu.txt @@ -1,4 +1,4 @@ -# Contains packages recommended for functional performance +# Contains packages requirements for GPU installation --find-links https://download.pytorch.org/whl/torch_stable.html torch==2.1.0+cu118 --find-links https://data.pyg.org/whl/torch-2.1.0+cu118.html