From d3c455a423730ace3cf2c0903be95e30c6678409 Mon Sep 17 00:00:00 2001 From: Tyler Date: Tue, 6 Dec 2022 08:02:35 -0500 Subject: [PATCH] Fix cuda runtime version --- Dockerfile.gpu | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/Dockerfile.gpu b/Dockerfile.gpu index 13d763d..87ce9e8 100644 --- a/Dockerfile.gpu +++ b/Dockerfile.gpu @@ -1,4 +1,4 @@ -FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu18.04 AS python-ubuntu +FROM nvidia/cuda:10.2-cudnn8-runtime-ubuntu18.04 AS python-ubuntu RUN apt-get update RUN apt-get install -y build-essential curl python3 python3-pip python3-venv @@ -10,7 +10,10 @@ RUN python3 -m pip install wheel FROM python-ubuntu -RUN apt-get update && apt-get install liblapack-dev libpng-dev libfreetype6-dev libqhull-dev git pkg-config portaudio19-dev swig libpulse-ocaml-dev gfortran libopenblas-dev libatlas-base-dev -y +RUN apt-get update && apt-get install -y \ + liblapack-dev libpng-dev libfreetype6-dev libqhull-dev git \ + pkg-config portaudio19-dev swig libpulse-ocaml-dev gfortran \ + libopenblas-dev libatlas-base-dev RUN mkdir /training && pip3 install precise-lite && pip3 install tensorflow-gpu==2.3.1