From 90b40394e8df298c6e5723f0f663d94d6dd3e01e Mon Sep 17 00:00:00 2001 From: Tyler Date: Tue, 6 Dec 2022 08:11:09 -0500 Subject: [PATCH] Use cudnn7 and force 10.1 for cuda --- Dockerfile.gpu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Dockerfile.gpu b/Dockerfile.gpu index 87ce9e8..ba8e04e 100644 --- a/Dockerfile.gpu +++ b/Dockerfile.gpu @@ -1,4 +1,4 @@ -FROM nvidia/cuda:10.2-cudnn8-runtime-ubuntu18.04 AS python-ubuntu +FROM nvidia/cuda:10.1-cudnn7-runtime-ubuntu18.04 AS python-ubuntu RUN apt-get update RUN apt-get install -y build-essential curl python3 python3-pip python3-venv