Skip to content

Commit afb72ca

Browse files
author
Aaron Boxer
committed
update setting LD_LIBRARY_PATH for cudnn 8.9.6.50
1 parent a863bff commit afb72ca

File tree

3 files changed

+4
-4
lines changed

3 files changed

+4
-4
lines changed

Dockerfile

-1
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,6 @@ RUN /bin/bash -c "\
2424

2525
# Set some environment variables
2626
ENV GST_PLUGIN_PATH=/root/gst-python-ml/plugins
27-
ENV LD_LIBRARY_PATH=/opt/venv/lib/python3.12/site-packages/nvidia/cublas/lib:/opt/venv/lib/python3.12/site-packages/nvidia/cudnn/lib
2827

2928
# allow Python to properly handle Unicode characters during logging.
3029
ENV PYTHONIOENCODING=utf-8

README.md

-3
Original file line numberDiff line numberDiff line change
@@ -35,8 +35,6 @@ or installing with a Docker container:
3535

3636
#### Install packages
3737

38-
Note: `nvidia-cuda-toolkit` dependency assumes an Nvidia card on your system
39-
4038
```
4139
sudo apt update && sudo apt -y upgrade
4240
sudo apt install -y python3-pip python3-venv \
@@ -61,7 +59,6 @@ sudo apt install -y python3-pip python3-venv \
6159
export VIRTUAL_ENV=$HOME/venv
6260
export PATH=$VIRTUAL_ENV/bin:$PATH
6361
export GST_PLUGIN_PATH=$HOME/src/gst-python-ml/plugins
64-
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$VIRTUAL_ENV/lib/python3.12/site-packages/nvidia/cublas/lib:$VIRTUAL_ENV/lib/python3.12/site-packages/nvidia/cudnn/lib
6562
```
6663

6764
and then

requirements/language_requirements.txt

+4
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,10 @@
22
# TTS and STT
33

44
nvidia-cudnn-cu11==8.9.6.50
5+
# to use this version of cuda, you must also update LD_LIBRARY_PATH in
6+
# ~/.bashrc with the following line (adjust for your python version)
7+
#export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$VIRTUAL_ENV/lib/python3.12/site-packages/nvidia/cublas/lib:$VIRTUAL_ENV/lib/python3.12/site-packages/nvidia/cudnn/lib
8+
59
sentencepiece
610
protobuf
711
accelerate

0 commit comments

Comments
 (0)