Dockerfile

FROM nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04
MAINTAINER Tom Breuel <[email protected]>
ENV DEBIAN_FRONTEND=noninteractive
RUN date && apt-get -qqy update

ENV TZ="America/Los Angeles"
RUN echo $TZ > /etc/timezone
RUN rm -f /etc/localtime && ln -snf /usr/share/zoneinfo/$TZ /etc/localtime || true

RUN apt-get install -qqy tzdata lsb-release openssh-client git gnupg \
    apt-transport-https curl iputils-ping dnsutils parallel \
    graphicsmagick imagemagick

RUN apt-get install -qqy vim-nox
RUN apt-get install -qqy python3 python3-pip
RUN pip3 install numpy scipy
# RUN pip3 install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
RUN pip3 install torch torchvision torchaudio
RUN pip3 install six future msgpack pyzmq simplejson braceexpand
RUN pip3 install torchtext==0.8.1
RUN pip3 install pytorch-lightning
RUN pip3 install simplejson braceexpand msgpack pyyaml Pillow
RUN pip3 install typer
RUN pip3 install --no-cache-dir --upgrade 'git+git://github.com/tmbdev/webdataset.git#webdataset' # 2
RUN pip3 install --no-cache-dir --upgrade 'git+git://github.com/nvlabs/tensorcom.git#tensorcom' # 2
RUN apt-get install -qqy mpi-default-bin
RUN pip3 install neptune-client
RUN pip3 install psutil