系统相关
首页 > 系统相关> > Windows10系统下利用Anaconda安装tensorflow-gpu

Windows10系统下利用Anaconda安装tensorflow-gpu

作者:互联网

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版本

标签:Anaconda,Windows10,cudnn,cuda,版本,gpu,tensorflow,安装
来源: https://www.cnblogs.com/lihanlihan/p/15863722.html