mirror of
https://github.com/jupyter/docker-stacks.git
synced 2025-10-18 15:32:56 +00:00
22 lines
741 B
Docker
22 lines
741 B
Docker
# Copyright (c) Jupyter Development Team.
|
|
# Distributed under the terms of the Modified BSD License.
|
|
ARG REGISTRY=quay.io
|
|
ARG OWNER=jupyter
|
|
ARG BASE_IMAGE=$REGISTRY/$OWNER/scipy-notebook
|
|
FROM $BASE_IMAGE
|
|
|
|
LABEL maintainer="Jupyter Project <jupyter@googlegroups.com>"
|
|
|
|
# Fix: https://github.com/hadolint/hadolint/wiki/DL4006
|
|
# Fix: https://github.com/koalaman/shellcheck/wiki/SC3014
|
|
SHELL ["/bin/bash", "-o", "pipefail", "-c"]
|
|
|
|
# Install PyTorch with pip (https://pytorch.org/get-started/locally/)
|
|
# hadolint ignore=DL3013
|
|
RUN pip install --no-cache-dir --index-url 'https://download.pytorch.org/whl/cpu' \
|
|
'torch' \
|
|
'torchaudio' \
|
|
'torchvision' && \
|
|
fix-permissions "${CONDA_DIR}" && \
|
|
fix-permissions "/home/${NB_USER}"
|