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
|
