piper-de-glados / cudnn_fix
systemofapwne's picture
transcribe container now properly uses GPU
bb61afd
raw
history blame
376 Bytes
# Fixes cudnn not working inside faster-whisper container of LSIO: https://github.com/linuxserver/docker-faster-whisper/blob/gpu/root/etc/s6-overlay/s6-rc.d/svc-whisper/run
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__))')