From 15742c852024767e03d9fc90cc8a3a577e54d817 Mon Sep 17 00:00:00 2001 From: Nalin Gupta Date: Thu, 14 Nov 2024 09:58:56 +0000 Subject: [PATCH] Changed image and removed tensorflow references --- jupyter-pytorch-notebook/Dockerfile | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/jupyter-pytorch-notebook/Dockerfile b/jupyter-pytorch-notebook/Dockerfile index f928990..3c99bd5 100644 --- a/jupyter-pytorch-notebook/Dockerfile +++ b/jupyter-pytorch-notebook/Dockerfile @@ -1,9 +1,9 @@ # Use a GPU-enabled base image based on the recommendation from # https://jupyter-docker-stacks.readthedocs.io/en/latest/using/selecting.html#gpu-accelerated-notebooks -# As this combines the tensorflow-notebook and the cuda-notebook, so we only +# As this combines the pytorch-notebook and the cuda-notebook, so we only # need to install the additional packages for each course. -FROM cschranz/gpu-jupyter:v1.5_cuda-11.6_ubuntu-20.04_python-only +FROM cschranz/gpu-jupyter:v1.7_cuda-12.3_ubuntu-22.04_python-only USER root # The base image only gets updated when a new version of CUDA is released. @@ -15,7 +15,6 @@ RUN apt-get update && \ USER ${NB_UID} -# Required for Tensorboard on a remote container RUN mamba install --yes \ 'jupyter-server-proxy' && \ # SciML requirements