From c5645d94edfdda8b716cdc0a7c5282267067c914 Mon Sep 17 00:00:00 2001 From: "Dutta, Ayushi (PG/R - Comp Sci & Elec Eng)" <ayushi.dutta@surrey.ac.uk> Date: Mon, 7 Apr 2025 16:26:51 +0000 Subject: [PATCH] runtime baase --- container.def | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/container.def b/container.def index 4c077b1..cfdcf43 100644 --- a/container.def +++ b/container.def @@ -1,6 +1,6 @@ # Get the base image from nvidia/cuda Docker image. BootStrap: docker -From: nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04 +From: nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04 %post # Install special software or additions @@ -20,8 +20,9 @@ From: nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04 export TORCH_CUDA_ARCH_LIST="7.5;8.0;8.6+PTX" export TCNN_CUDA_ARCHITECTURES="75;80;86" . /opt/conda/etc/profile.d/conda.sh # Non - interactive sessions we need this - conda init bash # For future interactive sessions, optional + mkdir -p /workspace + cd /workspace git clone https://github.com/snuvclab/gala.git cd gala conda env create -f env.yaml -- GitLab