Skip to content

Commit

Permalink
fix(whisper): failing gpu pod due to mangled env variable
Browse files Browse the repository at this point in the history
* Removes newlines that were causing the creation of an invalid environment variable entry.
* Resolves issue with failing gpu pod being unable to find the libcudnn_ops_infer.so file.
  • Loading branch information
CollectiveUnicorn authored Oct 3, 2024
1 parent 81ff6a3 commit bb5e58c
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions packages/whisper/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -37,8 +37,8 @@ COPY --from=builder /leapfrogai/.venv/ /leapfrogai/.venv/

# set the path to the cuda 11.8 dependencies
ENV LD_LIBRARY_PATH \
/leapfrogai/.venv/lib64/python3.11/site-packages/nvidia/cublas/lib:\
/leapfrogai/.venv/lib64/python3.11/site-packages/nvidia/cudnn/lib
/leapfrogai/.venv/lib64/python3.11/site-packages/nvidia/cublas/lib:\
/leapfrogai/.venv/lib64/python3.11/site-packages/nvidia/cudnn/lib

COPY packages/whisper/main.py .

Expand Down

0 comments on commit bb5e58c

Please sign in to comment.