Skip to content

Commit 56eab1a

Browse files
authored
Update the pip.md file on how to automatically add cuda/dnn path
Fixes tensorflow/tensorflow#60183 Modified the below lines CUDNN_PATH=$(dirname $(python -c "import nvidia.cudnn;print(nvidia.cudnn.__file__)")) export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib/:$CUDNN_PATH/lib with these lines of code mkdir -p $CONDA_PREFIX/etc/conda/activate.d echo 'CUDNN_PATH=$(dirname $(python -c "import nvidia.cudnn;print(nvidia.cudnn.__file__)"))' > $CONDA_PREFIX/etc/conda/activate.d/env_vars.sh echo 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib/:$CUDNN_PATH/lib' >> $CONDA_PREFIX/etc/conda/activate.d/env_vars.sh
1 parent f2427a6 commit 56eab1a

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

site/en/install/pip.md

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -28,8 +28,9 @@ step-by-step instructions.
2828
```bash
2929
conda install -c conda-forge cudatoolkit=11.8.0
3030
python3 -m pip install nvidia-cudnn-cu11==8.6.0.163 tensorflow==2.12.*
31-
CUDNN_PATH=$(dirname $(python -c "import nvidia.cudnn;print(nvidia.cudnn.__file__)"))
32-
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib/:$CUDNN_PATH/lib
31+
mkdir -p $CONDA_PREFIX/etc/conda/activate.d
32+
echo 'CUDNN_PATH=$(dirname $(python -c "import nvidia.cudnn;print(nvidia.cudnn.__file__)"))' > $CONDA_PREFIX/etc/conda/activate.d/env_vars.sh
33+
echo 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib/:$CUDNN_PATH/lib' >> $CONDA_PREFIX/etc/conda/activate.d/env_vars.sh
3334
# Verify install:
3435
python3 -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"
3536
```
@@ -75,8 +76,9 @@ step-by-step instructions.
7576
```bash
7677
conda install -c conda-forge cudatoolkit=11.8.0
7778
python3 -m pip install nvidia-cudnn-cu11==8.6.0.163 tensorflow==2.12.*
78-
CUDNN_PATH=$(dirname $(python -c "import nvidia.cudnn;print(nvidia.cudnn.__file__)"))
79-
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib/:$CUDNN_PATH/lib
79+
mkdir -p $CONDA_PREFIX/etc/conda/activate.d
80+
echo 'CUDNN_PATH=$(dirname $(python -c "import nvidia.cudnn;print(nvidia.cudnn.__file__)"))' > $CONDA_PREFIX/etc/conda/activate.d/env_vars.sh
81+
echo 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib/:$CUDNN_PATH/lib' >> $CONDA_PREFIX/etc/conda/activate.d/env_vars.sh
8082
# Verify install:
8183
python3 -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"
8284
```

0 commit comments

Comments
 (0)