Newer
Older
Bootstrap: docker
From: continuumio/miniconda3
# update conda and the OS
conda update conda && \
conda upgrade --all -y && \
apt-get update && \
apt-get -y install curl && \
DEBIAN_FRONTEND=noninteractive apt-get -y upgrade && \
DEBIAN_FRONTEND=noninteractive apt-get -y autoremove && \
DEBIAN_FRONTEND=noninteractive apt-get -y clean autoclean
conda activate py310-pytorch-1.13.1 && \
conda install -y pytorch pytorch-cuda=11.7 -c pytorch -c nvidia && \
conda list --explicit > /conda-list--explicit.yml
%post
eval "$(conda shell.bash hook)" && \
conda activate py310-pytorch-1.13.1 && \
"$@"