File tree Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change @@ -29,8 +29,9 @@ step-by-step instructions.
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
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
32
+ echo ' CUDNN_PATH=$(dirname $(python -c "import nvidia.cudnn;print(nvidia.cudnn.__file__)"))' >> $CONDA_PREFIX /etc/conda/activate.d/env_vars.sh
33
33
echo ' export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib/:$CUDNN_PATH/lib' >> $CONDA_PREFIX /etc/conda/activate.d/env_vars.sh
34
+ source $CONDA_PREFIX /etc/conda/activate.d/env_vars.sh
34
35
# Verify install:
35
36
python3 -c " import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"
36
37
```
@@ -77,8 +78,9 @@ step-by-step instructions.
77
78
conda install -c conda-forge cudatoolkit=11.8.0
78
79
python3 -m pip install nvidia-cudnn-cu11==8.6.0.163 tensorflow==2.12.*
79
80
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 ' CUDNN_PATH=$(dirname $(python -c "import nvidia.cudnn;print(nvidia.cudnn.__file__)"))' >> $CONDA_PREFIX /etc/conda/activate.d/env_vars.sh
81
82
echo ' export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib/:$CUDNN_PATH/lib' >> $CONDA_PREFIX /etc/conda/activate.d/env_vars.sh
83
+ source $CONDA_PREFIX /etc/conda/activate.d/env_vars.sh
82
84
# Verify install:
83
85
python3 -c " import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"
84
86
` ` `
You can’t perform that action at this time.
0 commit comments