Skip to content
Snippets Groups Projects
Commit 6a48c42f authored by Dutta, Ayushi (PG/R - Comp Sci & Elec Eng)'s avatar Dutta, Ayushi (PG/R - Comp Sci & Elec Eng)
Browse files

ld lib path u

parent c73df9e5
No related branches found
No related tags found
No related merge requests found
Pipeline #104101 canceled
......@@ -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.
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment