tensorflow 2.0 installation without explicit GPU

iot-salzburg · Jan 28, 2021 · 6a7acde · 6a7acde
1 parent cd12d58
commit 6a7acde
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/Dockerfile.gpulibs b/src/Dockerfile.gpulibs
@@ -3,7 +3,7 @@ LABEL maintainer="Christoph Schranz <[email protected]>"
 # Install Tensorflow, check compatibility here: https://www.tensorflow.org/install/gpu
 # installation via conda leads to errors in version 4.8.2
 RUN pip install --upgrade pip && \
-    pip install --no-cache-dir "tensorflow-gpu>=2.1.*" && \
+    pip install --no-cache-dir "tensorflow==2.3.2" && \
     pip install --no-cache-dir keras
 
 # Install PyTorch with dependencies