TensorFlow教程

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

这篇关于Windows10系统下利用Anaconda安装tensorflow-gpu的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!