tyler
/
precise-lite
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix cuda runtime version
continuous-integration/drone/push Build is passing Details

This commit is contained in:
Tyler 2022-12-06 08:02:35 -05:00
parent e22352a592
commit d3c455a423
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
Dockerfile.gpu
View File

@ -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 update
RUN apt-get install -y build-essential curl python3 python3-pip python3-venv 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 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 RUN mkdir /training && pip3 install precise-lite && pip3 install tensorflow-gpu==2.3.1