Skip to content
Snippets Groups Projects
Singularity 692 B
Newer Older
Bootstrap: docker
From: continuumio/miniconda3

Quang tru HUYNH's avatar
Quang tru HUYNH committed
%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

Quang tru HUYNH's avatar
Quang tru HUYNH committed
conda create --name py310-pytorch-1.13.1 -y && \
eval "$(conda shell.bash hook)"  && \
Quang tru HUYNH's avatar
Quang tru HUYNH committed
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
Quang tru HUYNH's avatar
Quang tru HUYNH committed
%post
eval "$(conda shell.bash hook)"  && \
conda activate py310-pytorch-1.13.1 && \
"$@"