From 19aed1a6c6cba09b582314e7a4a8943b1f00538d Mon Sep 17 00:00:00 2001
From: William Vale <wv00017@surrey.ac.uk>
Date: Wed, 26 Mar 2025 13:58:42 +0000
Subject: [PATCH] Edit container.def
---
container.def | 8 +++-----
1 file changed, 3 insertions(+), 5 deletions(-)
diff --git a/container.def b/container.def
index 5b13969..a8c2a8b 100644
--- a/container.def
+++ b/container.def
@@ -4,7 +4,7 @@ From: nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04
%files
# Copy the env.yaml file into the container
- requirements.txt /opt/env/requirements.txt
+ requirements.yml /opt/env/requirements.yml
%post
# Install special software or additions
@@ -27,8 +27,8 @@ From: nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04
/opt/conda/bin/conda init zsh
/opt/conda/bin/conda init tcsh
- # Install Python packages from MSOT_diffusion_env.yaml into the base environment
- /opt/conda/bin/conda env update -n base python=3.8 --prune
+ # Install Python packages from requirements.yaml into the base environment
+ /opt/conda/bin/conda env update -n base --file /opt/env/requirements.yaml --prune
%environment
# Set some environment vars
@@ -43,8 +43,6 @@ From: nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04
source /etc/profile.d/conda.sh
source ~/.bashrc
exec "$@"
- pip install "jax[cpu]===0.3.14" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html --use-deprecated legacy-resolver
- pip install -r /opt/env/requirements.txt
%labels
# Add custom metadata to the container.
--
GitLab