From 6a7acdecce5a66f036af66b7debf2fc30e5faa77 Mon Sep 17 00:00:00 2001 From: Christoph Schranz Date: Thu, 28 Jan 2021 11:17:16 +0100 Subject: [PATCH] tensorflow 2.0 installation without explicit GPU --- src/Dockerfile.gpulibs | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/Dockerfile.gpulibs b/src/Dockerfile.gpulibs index b933e05..e88157e 100644 --- a/src/Dockerfile.gpulibs +++ b/src/Dockerfile.gpulibs @@ -3,7 +3,7 @@ LABEL maintainer="Christoph Schranz " # 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