From f429c9a0fbd995a8b4454205ebd08437b1c150a5 Mon Sep 17 00:00:00 2001
From: Moritz Maxeiner <mm@ucw.sh>
Date: Tue, 1 Dec 2020 17:01:48 +0100
Subject: [PATCH] [jupyterhub] use gpu enabled tensorflow 2
---
jupyterhub/base/Dockerfile | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/jupyterhub/base/Dockerfile b/jupyterhub/base/Dockerfile
index 58cff49..e1f0002 100755
--- a/jupyterhub/base/Dockerfile
+++ b/jupyterhub/base/Dockerfile
@@ -93,7 +93,7 @@ RUN mkdir -p /usr/local/venv && \
python${PYTHON_VERSION} -m pip install --upgrade \
pip==${PYTHON_PIP_VERSION} \
setuptools==${PYTHON_SETUPTOOLS_VERSION} && \
- pip install tensorflow==2.3.1 && \
+ pip install tensorflow-gpu==2.3.1 && \
pip install \
wheel \
black \
--
GitLab