File tree Expand file tree Collapse file tree 2 files changed +9
-0
lines changed Expand file tree Collapse file tree 2 files changed +9
-0
lines changed Original file line number Diff line number Diff line change 19
19
f"{ READTHEDOCS_PROJECT } /envs/{ READTHEDOCS_VERSION } /bin/activate; ../../run setup read_the_docs'" ,
20
20
shell = True ,
21
21
)
22
+ # Update LD_LIBRARY_PATH for CUDA runtime
23
+ old_ld_library_path = os .environ .get ("LD_LIBRARY_PATH" , "" )
24
+ # Note that 'python3.8' is hard-coded here, it should be updated if the Python version changes by
25
+ # .readthedocs.yml or other configurations.
26
+ os .environ ["LD_LIBRARY_PATH" ] = (
27
+ f"{ READTHEDOCS_PROJECT } /envs/{ READTHEDOCS_VERSION } /lib/python3.8/site-packages/nvidia/cuda_runtime/lib:{ old_ld_library_path } "
28
+ )
22
29
subprocess .call (
23
30
"/bin/bash -c 'source /home/docs/checkouts/readthedocs.org/user_builds/"
24
31
f"{ READTHEDOCS_PROJECT } /envs/{ READTHEDOCS_VERSION } /bin/activate; ../../run setup_gen_docs'" ,
Original file line number Diff line number Diff line change @@ -306,6 +306,8 @@ install_python_dev_deps() {
306
306
# Read the Docs site is using specific setuptools version so should not upgrade it.
307
307
if [ " $config " = " read_the_docs" ]; then
308
308
run_command ${MONAI_PY_EXE} -m pip install -q -U pip wheel build
309
+ # Install cuda runtime dependency
310
+ run_command ${MONAI_PY_EXE} -m pip install -q nvidia-cuda-runtime-cu12
309
311
else
310
312
run_command ${MONAI_PY_EXE} -m pip install -q -U setuptools pip wheel build
311
313
fi
You can’t perform that action at this time.
0 commit comments