Hello_User/gpu-jupyter
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

tensorflow 2.0 installation without explicit GPU

Browse Source
This commit is contained in:
Christoph Schranz 2021-01-28 11:17:16 +01:00
parent cd12d58ab8
commit 6a7acdecce
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/Dockerfile.gpulibs
View File

@ -3,7 +3,7 @@ LABEL maintainer="Christoph Schranz <christoph.schranz@salzburgresearch.at>"
# 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