Skip to content
Snippets Groups Projects
Commit 19aed1a6 authored by William Vale's avatar William Vale
Browse files

Edit container.def

parent aa3746f0
No related branches found
No related tags found
No related merge requests found
Pipeline #100619 failed
......@@ -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.
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment