From e39a5f134a0a4ea84800163b71ce7ee081cb808d Mon Sep 17 00:00:00 2001 From: Drzony Date: Sun, 18 May 2025 00:59:27 +0200 Subject: [PATCH] Fix LD_LIBRARY_PATH for CUDA libraries. Closes #42 CUDA library directories do not have __init__.py files, so __file__ is None when importing them as modules. Using __path__ solves the problem. --- root/etc/s6-overlay/s6-rc.d/svc-whisper/run | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/root/etc/s6-overlay/s6-rc.d/svc-whisper/run b/root/etc/s6-overlay/s6-rc.d/svc-whisper/run index 5317dad..c8e8e50 100755 --- a/root/etc/s6-overlay/s6-rc.d/svc-whisper/run +++ b/root/etc/s6-overlay/s6-rc.d/svc-whisper/run @@ -1,7 +1,7 @@ #!/command/with-contenv bash # shellcheck shell=bash -export LD_LIBRARY_PATH=$(python3 -c 'import os; import nvidia.cublas.lib; import nvidia.cudnn.lib; print(os.path.dirname(nvidia.cublas.lib.__file__) + ":" + os.path.dirname(nvidia.cudnn.lib.__file__))') +export LD_LIBRARY_PATH=$(python3 -c 'import os; import nvidia.cublas.lib; import nvidia.cudnn.lib; print(os.path.dirname(nvidia.cublas.lib.__path__[0]) + "/lib:" + os.path.dirname(nvidia.cudnn.lib.__path__[0]) + "/lib")') exec \ s6-notifyoncheck -d -n 300 -w 1000 -c "nc -z localhost 10300" \