public_git/wyoming-piper-gpu
1
0
Fork 0
mirror of https://github.com/slackr31337/wyoming-piper-gpu.git synced 2026-10-19 07:51:59 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix LD_LIBRARY_PATH for CUDA libs

Browse Source
This commit is contained in:
Robert Dunmire III
2024-11-24 22:56:06 -05:00
parent bc8689954e
commit 9e3ec5426a
2 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Dockerfile
+5 -3
View File
@@ -41,14 +41,16 @@ RUN \
"wyoming-piper @ https://github.com/rhasspy/wyoming-piper/archive/refs/tags/v${WYOMING_PIPER_VERSION}.tar.gz" \
\
&& rm -r piper_phonemize-1.1.0-py3-none-any.whl \
\
RUN \
&& apt-get purge -y --auto-remove \
build-essential \
python3-dev \
\
&& rm -rf /var/lib/apt/lists/* /*.deb
&& rm -rf /var/lib/apt/lists/* /*.deb \
&& mkdir -p /share/piper
# Patch to enable CUDA in piper
# Patch to enable CUDA arguments for piper
COPY patch/process.py /usr/local/lib/python3.10/dist-packages/wyoming_piper/
COPY patch/__main__.py /usr/local/lib/python3.10/dist-packages/wyoming_piper/
run.sh
+3 -2
View File
@@ -1,9 +1,10 @@
#!/usr/bin/env bash
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/lib/python3.10/dist-packages/nvidia/cudnn/lib/
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/lib/python3.10/dist-packages/nvidia/cudnn/lib/:/usr/local/lib/python3.10/dist-packages/nvidia/cublas/lib:/usr/local/lib/python3.10/dist-packages/nvidia/curand/lib/:/usr/local/lib/python3.10/dist-packages/nvidia/cufft/lib/:/usr/local/lib/python3.10/dist-packages/nvidia/cuda_runtime/lib/:/usr/local/lib/python3.10/dist-packages/nvidia/cuda_nvrtc/lib/
python3 -m wyoming_piper \
--piper 'piper' \
--piper '/usr/local/bin/piper' \
--cuda \
--uri 'tcp://0.0.0.0:10200' \
--data-dir /data \
--data-dir /share/piper \
--download-dir /data "$@"