Skip to content

Commit 27eff62

Browse files
committed
fix the cannot find libnvrtc-builtins.so.13.0 issue
1 parent c341d98 commit 27eff62

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

.github/scripts/install-torch-tensorrt.sh

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,12 @@ pip uninstall -y torch torchvision
2121
pip install --force-reinstall --pre ${TORCHVISION} --index-url ${INDEX_URL} --extra-index-url https://pypi.org/simple
2222
pip install --force-reinstall --pre ${TORCH} --index-url ${INDEX_URL} --extra-index-url https://pypi.org/simple
2323

24+
# If CUDA 13 (cu13), prepend venv's NVIDIA CUDA 13 libs to LD_LIBRARY_PATH
25+
if [[ "${CU_VERSION}" == cu13* ]]; then
26+
SITE_PACKAGES="$(python -c 'import sysconfig; print(sysconfig.get_path("platlib"))')"
27+
export LD_LIBRARY_PATH="${SITE_PACKAGES}/nvidia/cu13/lib${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}"
28+
fi
29+
2430
if [[ ${USE_TRT_RTX} == true ]]; then
2531
source .github/scripts/install-tensorrt-rtx.sh
2632
# tensorrt-rtx is not publicly available, so we need to install the wheel from the tar ball

0 commit comments

Comments
 (0)