zhaosssss/torch_lab repository overview

FROM pytorch/pytorch:1.11.0-cuda11.3-cudnn8-devel
LABEL maintainer "zhaoshuai"

# RUN apt-key adv --keyserver keyserver.ubuntu.com --recv-keys A4B469963BF863CC
RUN rm /etc/apt/sources.list.d/cuda.list
RUN rm /etc/apt/sources.list.d/nvidia-ml.list
RUN apt-get update && apt-get install -y apt-transport-https \
              && apt-get install -y libturbojpeg libturbojpeg-dev

RUN \
	/opt/conda/bin/conda install -c fastai -c conda-forge -c anaconda -c conda-forge \
	av matplotlib scipy scikit-learn opencv-python-headless tensorboardx tensorboard einops \
	python-lmdb jupyter tqdm setproctitle pyturbojpeg seaborn h5py \
	&& /opt/conda/bin/conda clean -ya \
	&& pip install coolgpus

FROM zhaosssss/torch_lab:1.9.2
LABEL maintainer "zhaoshuai"
RUN pip install ftfy regex boto3

FROM zhaosssss/torch_lab:1.9.1
LABEL maintainer "zhaoshuai"
RUN conda install -c conda-forge librosa
RUN pip install pytorchvideo

FROM zhaosssss/torch_lab:1.9.0
LABEL maintainer "zhaoshuai"
RUN conda install -c conda-forge einops

FROM pytorch/pytorch:1.9.0-cuda11.1-cudnn8-devel
LABEL maintainer "zhaoshuai"
RUN apt-get update && apt-get install -y apt-transport-https \
					&& apt-get install -y libturbojpeg
RUN \
	/opt/conda/bin/conda install -c fastai -c conda-forge -c anaconda -c tensorly -c conda-forge -c fvcore -c iopath \
	tensorly cython matplotlib scipy scikit-learn opencv-python-headless cupy tensorboardx tensorboard fvcore=0.1.4 iopath \
	python-lmdb msgpack-python pyarrow jupyter tqdm setproctitle sphinx=3.2.0 av parameterized mock pyturbojpeg scipy scikit-learn \
	&& /opt/conda/bin/conda clean -ya \
	&& pip install coolgpus

ROM zhaosssss/torch_lab:1.8.0
LABEL maintainer "zhaoshuai"
RUN \
	conda install -c anaconda scipy scikit-learn \
	&& conda clean -ya \

FROM pytorch/pytorch:1.8.1-cuda11.1-cudnn8-devel
LABEL maintainer "zhaoshuai"
RUN apt-get update && apt-get install -y apt-transport-https \
					&& apt-get install -y libturbojpeg
RUN \
	/opt/conda/bin/conda install -c fastai -c conda-forge -c anaconda -c tensorly -c conda-forge -c fvcore -c iopath \
	tensorly cython matplotlib scipy scikit-learn opencv-python-headless cupy tensorboardx tensorboard fvcore=0.1.4 iopath \
	python-lmdb msgpack-python pyarrow jupyter tqdm setproctitle sphinx=3.2.0 av parameterized mock pyturbojpeg \
	&& /opt/conda/bin/conda clean -ya \
	&& pip install coolgpus

FROM zhaosssss/torch_lab:1.6.0x
LABEL maintainer "zhaoshuai"
RUN apt-get update && apt-get install -y apt-transport-https
RUN conda install -c conda-forge python-lmdb msgpack-python pyarrow

# Dockerfile of zhaosssss/torch_lab:1.6.0x
FROM pytorch/pytorch:1.6.0-cuda10.1-cudnn7-devel
LABEL maintainer "zhaoshuai"
RUN apt-get update && apt-get install -y apt-transport-https
RUN conda install -c fastai -c conda-forge -c anaconda -c tensorly -c conda-forge \
    tensorly cython matplotlib opencv-python-headless scipy scikit-learn cupy tensorboardx tensorboard
    # && conda clean -ya
RUN pip install --upgrade pip \
	&& pip install jupyter \
	&& pip install tqdm \
	&& pip install setproctitle

# Dockerfile of zhaosssss/torch_lab:1.6.0
FROM pytorch/pytorch:1.6.0-cuda10.1-cudnn7-devel
LABEL maintainer "zhaoshuai"

RUN pip install -i https://pypi.tuna.tsinghua.edu.cn/simple --upgrade pip \
        &&  pip install -i https://pypi.tuna.tsinghua.edu.cn/simple Cython \
        &&  pip install -i https://pypi.tuna.tsinghua.edu.cn/simple Pillow \
        &&  pip install -i https://pypi.tuna.tsinghua.edu.cn/simple opencv-python \
        &&  pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tqdm \
        &&  pip install -i https://pypi.tuna.tsinghua.edu.cn/simple setproctitle \
        &&  pip install -i https://pypi.tuna.tsinghua.edu.cn/simple pycocotools \
        &&  pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow \
        &&  pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorboardX \
        &&  pip install -i https://pypi.tuna.tsinghua.edu.cn/simple scipy \
        &&  pip install -i https://pypi.tuna.tsinghua.edu.cn/simple matplotlib \
        &&  pip install -i http://mirrors.aliyun.com/pypi/simple/ autotorch --pre \
        &&  pip install -i https://pypi.tuna.tsinghua.edu.cn/simple torch-encoding --pre \
        &&  pip install -i https://pypi.tuna.tsinghua.edu.cn/simple albumentations