From 9e3ec5426a5273f84be7e2efabb6fd9bd71f0f04 Mon Sep 17 00:00:00 2001
From: Robert Dunmire III <rdunmire@i-evolve.com>
Date: Sun, 24 Nov 2024 22:56:06 -0500
Subject: [PATCH] Fix LD_LIBRARY_PATH for CUDA libs

---
 Dockerfile | 8 +++++---
 run.sh     | 5 +++--
 2 files changed, 8 insertions(+), 5 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index 1fb385b..81b1c08 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -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/
 
diff --git a/run.sh b/run.sh
index d81a500..e9592fc 100644
--- a/run.sh
+++ b/run.sh
@@ -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 "$@"