From 104b23ffc9a91c14720bcf5748772222b6aed619 Mon Sep 17 00:00:00 2001
From: "Dutta, Ayushi (PG/R - Comp Sci & Elec Eng)" <ayushi.dutta@surrey.ac.uk>
Date: Fri, 11 Apr 2025 12:44:02 +0000
Subject: [PATCH] inclued cuda home
---
container.def | 2 ++
1 file changed, 2 insertions(+)
diff --git a/container.def b/container.def
index 78252b6..697ee4d 100644
--- a/container.def
+++ b/container.def
@@ -14,6 +14,8 @@ From: nvidia/cuda:11.7.1-cudnn8-devel-ubuntu22.04
# Set some environment vars
export TORCH_CUDA_ARCH_LIST="7.5;8.0;8.6+PTX"
export TCNN_CUDA_ARCHITECTURES="75;80;86"
+ export CUDA_HOME=/usr/local/cuda
+ export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$CUDA_HOME/lib64/stubs:$LD_LIBRARY_PATH
%labels
# Add custom metadata to the container.
--
GitLab