diff --git a/jupyterhub/base/Dockerfile b/jupyterhub/base/Dockerfile index 58cff4933f8455d4cdc4f3e6356177619d8e7654..e1f00025446c4d5594666c54a1a38b2033468217 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 \