Skip to content

Commit

Permalink
Changed image and removed tensorflow references
Browse files Browse the repository at this point in the history
  • Loading branch information
Neonbluestoplight committed Nov 14, 2024
1 parent d9ce834 commit 15742c8
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions jupyter-pytorch-notebook/Dockerfile
Original file line number Diff line number Diff line change
@@ -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.
Expand All @@ -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
Expand Down

0 comments on commit 15742c8

Please sign in to comment.