From 06313b210a6b9335dcf9e75b8ac9b4155dc6e19c Mon Sep 17 00:00:00 2001 From: Rasmus Oersoe Date: Mon, 9 Oct 2023 13:15:17 +0200 Subject: [PATCH] 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