Install tensorflow-gpu earlier in Dockerfile

HULKs · Jul 2, 2021 · 22013bf · 22013bf
1 parent d6f8b46
commit 22013bf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/worker/genetic_individual_neuralnet/Dockerfile b/worker/genetic_individual_neuralnet/Dockerfile
@@ -18,7 +18,7 @@ RUN git clone https://github.com/boostorg/boost.git \
     && ./bootstrap.sh --prefix=/usr \
     && ./b2 install
 
-RUN pip install git+https://github.com/h3ndrk/PyCompiledNN.git tensorflow_addons
+RUN pip install git+https://github.com/h3ndrk/PyCompiledNN.git tensorflow-gpu tensorflow_addons
 
 COPY genetic_individual_neuralnet/ /genetic_individual_neuralnet/
 COPY worker/ /worker/