From 6d8539c6ea48bfa795f53554968f5c77f9d88551 Mon Sep 17 00:00:00 2001
From: Robert Sachunsky <38561704+bertsky@users.noreply.github.com>
Date: Tue, 15 Oct 2024 23:22:48 +0200
Subject: [PATCH] deps-cuda: make sure cudnn gets installed, pin torch version
 for CUDA 11
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

(sry, must commit to master directly – only Docker deploy is affected; current version is broken for all CUDA images)
---
 Makefile | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/Makefile b/Makefile
index 1708caa12..047772734 100644
--- a/Makefile
+++ b/Makefile
@@ -97,7 +97,7 @@ deps-cuda:
 # works, too:
 	shopt -s nullglob; \
 	$(PIP) install nvidia-pyindex \
-	 && $(PIP) install nvidia-cudnn-cu11==8.7.* \
+	 && $(PIP) install nvidia-cudnn-cu11~=8.7 \
 	                   nvidia-cublas-cu11~=11.11 \
 	                   nvidia-cusparse-cu11~=11.7 \
 	                   nvidia-cusolver-cu11~=11.4 \
@@ -158,7 +158,7 @@ deps-tf2:
 	fi
 
 deps-torch:
-	$(PIP) install -i https://download.pytorch.org/whl/cu118 torch torchvision
+	$(PIP) install -i https://download.pytorch.org/whl/cu118 torchvision==0.16.2+cu118 torch==2.1.2+cu118
 
 # Dependencies for deployment in an ubuntu/debian linux
 deps-ubuntu: