【解决】Tensorflow检测不到GPU

【解决】Tensorflow检测不到GPU

问题

import tensorflow as tf
tf.test.is_gpu_available()

结果为Flase

原因

由于TensorFlow 2.11之前的版本对window版本TensorFlow的GPU不支持了
所以安装版本应为

pip install "tensorflow<2.11"

解决

1. 安装NVIDIA的官方驱动 官方驱动
   

2. 下载Cuda11(重要),因为TensorFlow的2.10.0版本支持的cuda为11下载地址

3. 下载cuDNN下载链接解压后复制这三个文件夹
   
   放置于C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1文件夹下
   

4. 重启电脑

5. 安装

conda install cudatoolkit=11.1
conda install "tensorflow<2.11"

6. 测试

import tensorflow as tf
tf.config.list_physical_devices('GPU')

看到输出为

[PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU')]

成功！！