在Win10/Win11 WSL 中安装Tensorflow 2.12 GPU

基本安装这个链接做就可以:

https://www.tensorflow.org/install/pip

但是我一步一步做下来,最后每次

conda activate tf

之后,使用:

python3 -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"

并不能自动找到GPU。

最后发现我们还是需要自己修改一下文件:

$CONDA_PREFIX/etc/conda/activate.d/env_vars.sh

可以用任何编辑器(vim,gedit)打开,如果你按前面的教程做了,那么它的内容应该是:

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib/:$CUDNN_PATH/lib

在前面加一行:

CUDNN_PATH=$(dirname $(python -c "import nvidia.cudnn;print(nvidia.cudnn.__file__)"))

保存,退出。

之后每次使用

conda activate tf

再测试:

python3 -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"

都可以找到GPU了。

你可能感兴趣的:(tensorflow,深度学习,python)