File tree Expand file tree Collapse file tree 1 file changed +6
-4
lines changed Expand file tree Collapse file tree 1 file changed +6
-4
lines changed Original file line number Diff line number Diff line change @@ -28,8 +28,9 @@ step-by-step instructions.
28
28
``` bash
29
29
conda install -c conda-forge cudatoolkit=11.8.0
30
30
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
33
34
# Verify install:
34
35
python3 -c " import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"
35
36
```
@@ -75,8 +76,9 @@ step-by-step instructions.
75
76
` ` ` bash
76
77
conda install -c conda-forge cudatoolkit=11.8.0
77
78
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
80
82
# Verify install:
81
83
python3 -c " import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"
82
84
` ` `
You can’t perform that action at this time.
0 commit comments