From 25d980b3aab58e00fab0682426db6e96af6bc94f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Antonio=20Jes=C3=BAs?= Date: Thu, 16 Nov 2023 10:49:15 +0100 Subject: [PATCH] feat (JupyterHub): Changed to ubuntu cuda 12 image. --- jupyterhub/tensorflow-gpu/Dockerfile | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/jupyterhub/tensorflow-gpu/Dockerfile b/jupyterhub/tensorflow-gpu/Dockerfile index da01057..a6147e6 100644 --- a/jupyterhub/tensorflow-gpu/Dockerfile +++ b/jupyterhub/tensorflow-gpu/Dockerfile @@ -1,8 +1,9 @@ # Base image: https://github.com/jupyter/docker-stacks/blob/74bbd0bffc3b444e2d65279739bc2d681b6199e2/images/docker-stacks-foundation/Dockerfile # ARG ROOT_CONTAINER=tensorflow/tensorflow:2.15.0-gpu -ARG ROOT_CONTAINER=pytorch/pytorch:2.1.1-cuda12.1-cudnn8-runtime +# ARG ROOT_CONTAINER=pytorch/pytorch:2.1.1-cuda12.1-cudnn8-runtime # ARG ROOT_CONTAINER=ubuntu:22.04 +ARG ROOT_CONTAINER=nvidia/cuda:12.2.2-cudnn8-runtime-ubuntu22.04 ARG FOUNDATION_CONTAINER=quay.io/jupyter/docker-stacks-foundation:4d70cf8da953 ARG BASE_NOTEBOOK=quay.io/jupyter/base-notebook:b86753318aa1 @@ -121,7 +122,8 @@ RUN pip install jupyter-core \ jupyterlab \ notebook \ jupyterhub \ - nbclassic && \ + nbclassic \ + tensorflow && \ jupyter server --generate-config && \ jupyter lab clean && \ rm -rf "/home/${NB_USER}/.cache/yarn" && \