安装Tensorflow及GPU支持

• 安装基础库

  • https://tensorflow.google.cn/install/gpu?hl=zh-cn
  • 【2021-09更新】安装CUDA11.4。 11.2的“无法找到 cusolver64_10.dll” 的错误已解决。详情见：https://github.com/tensorflow/tensorflow/issues/44291
    • https://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/
  • 安装cuDNN 11.4
    • 下载并解压。

• 设置系统变量

  • PATH：
    • C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4\bin
    • C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4\extras\CUPTI\lib64
    • C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4\include
    • C:<cuDNN安装目录>\bin

• 验证CUDA11.4 安装

  • 在CUDA安装目录（C:\ProgramData\NVIDIA Corporation\CUDA Samples\v11.4\1_Utilities\deviceQuery）中，使用VS2019打开deviceQuery_vs2019.sln。点击生成（build）
  • 在生成目录（C:\ProgramData\NVIDIA Corporation\CUDA Samples\v11.4\bin\win64\Debug），使用CMD运行deviceQuery.exe

• 验证Tensorflow能使用GPU

  import tensorflow as tf
  tf.test.is_gpu_available() or tf.config.list_physical_devices('GPU')