12 lines
360 B
Docker
12 lines
360 B
Docker
|
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
|
||
|
|
RUN pip3 -q install pip --upgrade
|
||
|
|
|
||
|
|
RUN pip3 install jupyter numpy pandas torch torchvision tensorboardX
|
||
|
|
|
||
|
|
FROM jupyter-base
|
||
|
|
RUN pip3 install transformers barbar matplotlib
|