Windows系统让tensorflow找到gpu

等待这一天太久太久了，翻来覆去笨蛋tf只能找到cpu，我终于在一系列应用降级后让他看到了我的gpuT.T…

CUDA 配置

1. 要求：

   • NVIDIA® GPU 驱动程序 - CUDA® 11.2
     这个基本上现在电脑人人都有，在powershell终端输入nvidia-smi查看。
   • CUDA® 工具包 - CUDA toolkit 11.2
     进入这个链接，找到CUDA Toolkit 11.2.0 (December 2020), Versioned Online Documentation下载，型号必须对。
     我是win11，但是选择window-x86_64-version10下载没有问题。
     注意
     如果之前下过更新版本的cuda工具包，会报错you already have newer version..., 先去windows设置–>应用–>应用和功能，把带cuda字样的应用都删掉，NVIDIA Frameview SDK也要删！然后再安装。
     安装后在powershell终端使用nvcc -V确认。
   • CUDA® 工具包附带的 CUPTI
     这个自动跟随cuda toolkit安装，不用管。
   • cuDNN SDK - 8.1.0 版本
     使用这个链接，注册以后选择Download cuDNN v8.1.0 (January 26th, 2021), for CUDA 11.0,11.1 and 11.2下载，别的不行。
     下载完是一个压缩包，解压到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2，解压后的文件夹叫cuda，重命名为cudnn。
   2. 环境配置
      进入windows的编辑环境变量，将
      C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2\libnvvp
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2\cudnn\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2\extras\CUPTI\lib64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2\include
      加入PATH。

下载tensorflow

tensorflow对于gpu的支持只到2.10，如果你装了最新的tf(2.11)。
先卸载

pip uninstall keras
pip uninstall tensorflow-io-gcs-filesystem
pip uninstall tensorflow-estimator
pip uninstall tensorflow
pip uninstall Keras-Preprocessing
pip uninstall tensorflow-intel

然后

pip install tensorflow<2.11

即可。

验证

创建一个python文件

import tensorflow as tf
print(tf.config.list_physical_devices())

打印输出为[PhysicalDevice(name='/physical_device:CPU:0', device_type='CPU'), PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU')]。

Reference

1. https://www.tensorflow.org/install/gpu
2. https://forums.developer.nvidia.com/t/nvidia-installer-cannot-continue/169854/5
3. https://zhuanlan.zhihu.com/p/102966512
4. https://stackoverflow.com/questions/41402409/tensorflow-doesnt-seem-to-see-my-gpu/74662689#74662689