Skip to content
Snippets Groups Projects
Commit f429c9a0 authored by calrama's avatar calrama
Browse files

[jupyterhub] use gpu enabled tensorflow 2

parent 934122fb
Branches
No related tags found
No related merge requests found
Pipeline #33388 passed
...@@ -93,7 +93,7 @@ RUN mkdir -p /usr/local/venv && \ ...@@ -93,7 +93,7 @@ RUN mkdir -p /usr/local/venv && \
python${PYTHON_VERSION} -m pip install --upgrade \ python${PYTHON_VERSION} -m pip install --upgrade \
pip==${PYTHON_PIP_VERSION} \ pip==${PYTHON_PIP_VERSION} \
setuptools==${PYTHON_SETUPTOOLS_VERSION} && \ setuptools==${PYTHON_SETUPTOOLS_VERSION} && \
pip install tensorflow==2.3.1 && \ pip install tensorflow-gpu==2.3.1 && \
pip install \ pip install \
wheel \ wheel \
black \ black \
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment