Ubuntu 20, GPU supported.
FROM nvidia/cuda:11.4.0-base-ubuntu20.04
RUN apt-key adv --fetch-keys https://developer.download.nvidia.cn/compute/cuda/repos/ubuntu2004/x86_64/7fa2af80.pub
RUN apt-get update && \
apt-get install -y python3 python3-pip vim && \
rm -rf /var/lib/apt/lists/*
RUN pip3 install codefast -i https://pypi.douban.com/simple \
flask -i https://pypi.douban.com/simple
curl and iproute2 (ip addr to get address) installed on Version 20230315
what is new?
RUN pip3 install torch torchvision torchaudio transformer sentencepiece gunicorn
Content type
Image
Digest
sha256:20548e4c3…
Size
4.6 GB
Last updated
over 3 years ago
docker pull signalout/cuda_python38:transformers