Anaconda+tensorflow-gpu2.6.0+python3.7+cuda11.2+cudnn8.1
一、步骤概述
1、查看电脑对应的cuda版本
2、查看所需的tensorflow-gup版本对应的cuda版本和cudnn版本(请在安装前一定要注意tensorflow-gpu,cuda和cudnn版本之间的对照关系,非常重要,必须一致!)
3、安装cuda和cudnn
4、安装Anaconda
5、安装tensorflow-gpu
6、检查tensorflow是否是gpu版本
二、详细流程
1、查看电脑对应的cuda版本
(1)首先,电脑必须有NVIDIA的显卡(其他类型的显卡不能使用cuda)
(2)鼠标右键->打开NVIDIA控制面板->检查对应的cuda版本
如图,CUDA 11.4.94即是NVIDIA驱动支持的最高版本的CUDA(向下兼容)
2、查看所需的tensorflow-gup版本对应的cuda版本和cudnn版本
请到该链接地址查看:https://tensorflow.google.cn/install/source_windows
3、安装cuda和cudnn
具体请参考:深度学习环境搭建(GPU)CUDA安装(完全版)
4、安装Anaconda
具体请参考:深度学习环境配置3——windows下的tensorflow-gpu=2.2.0环境配置
5、安装tensorflow-gpu
具体请参考:深度学习环境配置3——windows下的tensorflow-gpu=2.2.0环境配置
6、检查tensorflow是否是gpu版本
具体请参考:如何查看tensorflow跑的是gpu版本还是cpu版本