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

Use fork for gpu
All checks were successful
continuous-integration/drone/push Build is passing
Details

Browse Source
This commit is contained in:
Tyler 2022-12-07 00:53:18 -05:00
parent cc9b2b8a09
commit bf3131a13d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Dockerfile.gpu
View File

@ -14,7 +14,7 @@ RUN apt-get update && apt-get install -y \
pkg-config portaudio19-dev swig libpulse-ocaml-dev gfortran \
libopenblas-dev libatlas-base-dev
RUN mkdir /training && pip3 install precise-lite && pip3 install tensorflow==2.4.4 && pip3 install tensorflow-gpu==2.4.4
RUN mkdir /training && pip3 install git+https://github.com/tystuyfzand/precise-lite.git@feature/tf24 && pip3 install tensorflow-gpu==2.4.4
WORKDIR /training