Use fork for gpu
continuous-integration/drone/push Build is passing
Details
continuous-integration/drone/push Build is passing
Details
This commit is contained in:
parent
cc9b2b8a09
commit
bf3131a13d
|
@ -14,7 +14,7 @@ RUN apt-get update && apt-get install -y \
|
||||||
pkg-config portaudio19-dev swig libpulse-ocaml-dev gfortran \
|
pkg-config portaudio19-dev swig libpulse-ocaml-dev gfortran \
|
||||||
libopenblas-dev libatlas-base-dev
|
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
|
WORKDIR /training
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue