diff --git a/container.def b/container.def
index 78252b6ae9900e32ae547fda721a6464ce7c6fda..697ee4d3192d9293f7f469ce9077124322b07aa4 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.