Install tensorflow-gpu
continuous-integration/drone/push Build is passing Details

This commit is contained in:
Tyler 2022-12-06 07:36:15 -05:00
parent 11943a2012
commit e22352a592
1 changed files with 1 additions and 1 deletions

View File

@ -12,7 +12,7 @@ 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 mkdir /training && pip3 install precise-lite
RUN mkdir /training && pip3 install precise-lite && pip3 install tensorflow-gpu==2.3.1
WORKDIR /training