diff --git a/Singularity b/Singularity index 0adbc0e48a7b3f1e6ccbda6ad736bfbfe4fc9774..140e745c0fb5c19921510582a29d7c2df2ac0d70 100644 --- a/Singularity +++ b/Singularity @@ -17,7 +17,7 @@ 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 -%post +%runscript eval "$(conda shell.bash hook)" && \ conda activate py310-pytorch-nightly && \ "$@"