From 22013bf3e691863b35898d7765f7a87200c85d62 Mon Sep 17 00:00:00 2001 From: Hendrik Date: Fri, 2 Jul 2021 23:21:37 +0200 Subject: [PATCH] Install tensorflow-gpu earlier in Dockerfile --- worker/genetic_individual_neuralnet/Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/worker/genetic_individual_neuralnet/Dockerfile b/worker/genetic_individual_neuralnet/Dockerfile index d9094b2..e006f34 100644 --- 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/