stud-ai/0-install/jupyter.Dockerfile

13 lines
379 B
Docker
Raw Permalink Normal View History

2020-12-02 13:22:03 +01:00
FROM nvidia/cuda:10.2-runtime AS jupyter-base
WORKDIR /
RUN apt update && apt install -y --no-install-recommends \
git build-essential \
2024-08-06 11:37:45 +02:00
python3-dev python3-pip python3-setuptools \
imagemagick
2020-12-02 13:22:03 +01:00
RUN pip3 -q install pip --upgrade
RUN pip3 install jupyter numpy pandas torch torchvision tensorboardX
FROM jupyter-base
RUN pip3 install transformers barbar matplotlib