diff --git a/container.def b/container.def
index a8ebc7bdec51ed45a64e0a49ad5e339403fd3c3b..7fa07d546027e67eab5c3d6fe91fe72799168a8c 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.