File tree Expand file tree Collapse file tree 1 file changed +6
-0
lines changed
Expand file tree Collapse file tree 1 file changed +6
-0
lines changed Original file line number Diff line number Diff line change @@ -21,6 +21,12 @@ pip uninstall -y torch torchvision
2121pip install --force-reinstall --pre ${TORCHVISION} --index-url ${INDEX_URL} --extra-index-url https://pypi.org/simple
2222pip 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+
2430if [[ ${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
You can’t perform that action at this time.
0 commit comments