TensorFlow2.0 学习笔记三 在GPU上训练，win11安装tensorflow2 GPU版本

官方参考

https://www.tensorflow.org/tutorials/keras/classification?hl=zh-cnGPU 支持  |  TensorFlow (google.cn)https://www.tensorflow.org/tutorials/keras/classification?hl=zh-cn

基本上按照上面来的都是不成功。。。。。。

安装软件和驱动

显卡驱动安装

安装对应显卡的最新驱动，最好选择Studio Driver类型

网址： Official Drivers | NVIDIAOfficial Drivers | NVIDIA

CUDA Toolkit安装

首先需要确认下显卡支持的版本，打开显卡控制面板查看信息，现在我这个支持最高12.2.136，则安装版本不超过这个

网址：CUDA Toolkit Archive | NVIDIA Developer

根据我的硬件选择了最新的

细节注意下，自定义安装，只需要CUDA即可

安装cuDNN

根据上面安装的CUDA，选择和自己版本相符的，我直接选择最新的8.9.3；

网址：cuDNN Archive | NVIDIA Developer

注意：如果是第一次进入，可能需要填写邮箱什么的

下载完成后，解压出来，需要把里面的include、lib、bin直接复制进安装目录

每个人目录可能不同C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.2

测试安装是否正常

进入路径C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.2\extras\demo_suite

其中版本可能不同，路径可能不同，打开PowerShell，运行bandwidthTest.exe和deviceQuery.exe均能看到信息，并且最终返回有Result = PASS

如果失败，最好再安装下VS，并且安装C++开发环境

安装anaconda

较简单，各自安装

Python和tensorflow环境搭建

创建Python3.9环境

使用3.9.x版本，这是官网指定版本，其他版本暂未测试

模块安装

解释下 为什么要使用conda安装，主要是想要他自动安装符合版本需求的包，避免冲突，这也是能成功运行的关键；

下面这些最好一个一个装

tensorflow选择gpu版本，2.6

keras，选择gpu版本 2.6

其他

在运行过程中可能会提示有些包版本还是不对，需要更换下对应的版本

from: TensorFlow2.0 学习笔记三 在GPU上训练，win11安装tensorflow2 GPU版本-优快云博客