Singularity

Bootstrap: docker
From: continuumio/miniconda3

%post
# 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 create --name py310-pytorch-nightly -y && \
eval "$(conda shell.bash hook)"  && \
conda activate py310-pytorch-nightly  && \
conda install -y pytorch pytorch-cuda=11.7 -c pytorch-nightly -c nvidia && \
conda list --explicit > /conda-list--explicit.yml

%runscript
eval "$(conda shell.bash hook)"  && \
conda activate py310-pytorch-nightly && \
"$@"