From 1994e05940c2949ff6c94593f57fab7932ced438 Mon Sep 17 00:00:00 2001 From: lrlunin Date: Mon, 27 Nov 2023 17:07:17 +0100 Subject: [PATCH] install torch, tf and cupy --- VERSION | 2 +- mbi-div-b-notebook-cuda/Dockerfile | 19 ++++++++++++++++++- 2 files changed, 19 insertions(+), 2 deletions(-) diff --git a/VERSION b/VERSION index 0d91a54..9e11b32 100644 --- a/VERSION +++ b/VERSION @@ -1 +1 @@ -0.3.0 +0.3.1 diff --git a/mbi-div-b-notebook-cuda/Dockerfile b/mbi-div-b-notebook-cuda/Dockerfile index d66a177..ad6b13f 100644 --- a/mbi-div-b-notebook-cuda/Dockerfile +++ b/mbi-div-b-notebook-cuda/Dockerfile @@ -146,7 +146,24 @@ RUN mkdir /opt/boris && \ USER ${NB_UID} -RUN fix-permissions "${CONDA_DIR}" && \ +RUN mamba install --yes \ + # due to the bug in previos version of notebook need to be fixed with the version below + -c 'nvidia' \ + -c 'pytorch' \ + 'cupy' \ + 'pytorch' \ + 'torchvision' \ + 'torchaudio' \ + 'pytorch-cuda=12.1' && \ + mamba clean --all -f -y && \ + fix-permissions "${CONDA_DIR}" && \ + fix-permissions "/home/${NB_USER}" + +# Install packages missing in mamba with pip +# ultrafastfitfunctions need to be installed from git and commit after accepting the pull request +# https://github.com/EmCeBeh/ultrafastFitFunctions/pull/2 +RUN pip install --no-cache-dir tensorflow[and-cuda] && \ + fix-permissions "${CONDA_DIR}" && \ fix-permissions "/home/${NB_USER}" USER ${NB_UID}