13 lines
379 B
Docker
Executable File
13 lines
379 B
Docker
Executable File
FROM nvidia/cuda:10.2-runtime AS jupyter-base
|
|
WORKDIR /
|
|
RUN apt update && apt install -y --no-install-recommends \
|
|
git build-essential \
|
|
python3-dev python3-pip python3-setuptools \
|
|
imagemagick
|
|
RUN pip3 -q install pip --upgrade
|
|
|
|
RUN pip3 install jupyter numpy pandas torch torchvision tensorboardX
|
|
|
|
FROM jupyter-base
|
|
RUN pip3 install transformers barbar matplotlib
|