From 6a48c42f2b94f9d6ac1dfe377505b29ef58f1ed9 Mon Sep 17 00:00:00 2001
From: "Dutta, Ayushi (PG/R - Comp Sci & Elec Eng)" <ayushi.dutta@surrey.ac.uk>
Date: Thu, 17 Apr 2025 07:54:33 +0000
Subject: [PATCH] ld lib path u
---
container.def | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/container.def b/container.def
index a8ebc7b..7fa07d5 100644
--- a/container.def
+++ b/container.def
@@ -18,7 +18,7 @@ From: nvidia/cuda:12.1.0-cudnn8-devel-ubuntu20.04
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=/usr/lib/x86_64-linux-gnu:$CUDA_HOME/lib64:$CUDA_HOME/lib64/stubs:$CONDA_PREFIX/lib:$LD_LIBRARY_PATH
+ export LD_LIBRARY_PATH=/usr/lib/x86_64-linux-gnu:$CUDA_HOME/lib64:$CUDA_HOME/lib64/stubs:$CONDA_PREFIX/lib:$CONDA_PREFIX/lib/python3.10/site-packages/torch/lib:$LD_LIBRARY_PATH
%labels
# Add custom metadata to the container.
--
GitLab