diff --git a/Dockerfile b/Dockerfile index 8b8e489589ac785e89459946882e89507b99c060..4095ceaaf037f6ddfdbf54a0f48a81ea26b7a8d9 100644 --- a/Dockerfile +++ b/Dockerfile @@ -9,13 +9,8 @@ RUN conda update conda && \ DEBIAN_FRONTEND=noninteractive apt-get -y autoremove && \ DEBIAN_FRONTEND=noninteractive apt-get -y clean autoclean -# download the yml file -RUN curl https://gitlab.pasteur.fr/tru/miniconda3-python310-pytorch-cuda11.7/-/raw/main/conda-env-export.yml > conda-env-export.yml && \ - conda create --name py310-pytorch --file conda-env-export.yml && \ +RUN conda create --name py310-pytorch-1.13.1 -y && \ eval "$(conda shell.bash hook)" && \ - conda activate py310-pytorch && \ - conda list --explicit > /tmp/conda-list--explicit.yml - - - - + 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 diff --git a/Singularity b/Singularity index 2856df158730026213b2b326bcfdc0eff1ea2d47..e7d1f39ad39c518dda74199546ac63fe2d348b7f 100644 --- a/Singularity +++ b/Singularity @@ -11,10 +11,13 @@ DEBIAN_FRONTEND=noninteractive apt-get -y upgrade && \ DEBIAN_FRONTEND=noninteractive apt-get -y autoremove && \ DEBIAN_FRONTEND=noninteractive apt-get -y clean autoclean -# download the yml file -curl https://gitlab.pasteur.fr/tru/miniconda3-python310-pytorch-cuda11.7/-/raw/main/conda-env-export.yml > conda-env-export.yml && \ -conda create --name py310-pytorch --file conda-env-export.yml && \ +conda create --name py310-pytorch-1.13.1 -y && \ eval "$(conda shell.bash hook)" && \ -conda activate py310-pytorch && \ -conda list --explicit > /tmp/conda-list--explicit.yml +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 && \ +"$@"