当前位置:   article > 正文

Tensorflow、CUDA、cuDNN详细的下载安装过程_cudnn下载

cudnn下载


前言

此篇文章所安装的Tensorflow版本为2.4.0,CUDA版本为11.0,cuDNN版本为8.0,其他的版本也可以参考,适合第一次装安装tensorflow的小伙伴学习。


一、Tensorflow简介

Tensorflow是目前机器学习主流框架以及其它涉及大量数学运算的算法库之一。Tensorflow由Google开发,是GitHub上最受欢迎的机器学习库之一。
在这里插入图片描述

二、Tensorflow版本介绍

1.Tensorflow与python、CUDA、cuDNN对应关系

首先进入Tensorflow官网:https://tensorflow.google.cn/install/source_windows
找到windows对应的版本的模块
在这里插入图片描述
在这里插入图片描述
然后再查看自己python的版本,在cmd中输入python --version查看python的版本,如果没有添加环境变量,找到所安装的python.exe,将其添加到环境变量中。
我的python版本是3.8,所以装tensorflow的话就装2.2-2.4
在这里插入图片描述
在这里插入图片描述
我们看到cuDNN、CUDA有不同的版本要求,但是Compiler——MSVC 2019(Microsoft Visual Studio 2019 BuildTools)、Build tools这两个需不需要安装呢?这个我暂时也不知道,我是新装的电脑,没装Compiler和Build tools也能正常运行tensorflow,所以暂且不管。
我们现在已经知道了Tensorflow与python、CUDA、cuDNN对应关系,还有一点需要注意——查看自己的电脑能装什么样的CUDA!

2.NVIDIA显卡和CUDA的版本号关系

官网查看地址:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html#title-new-features
在这里插入图片描述
我的是windows10,所以看到左右边的一列,这些数字代表Driver Version,下一步就是查看自己的电脑版本。
方法一:进入到nvidia-smi对应的文件夹,在地址栏里清空并输入cmd,然后输入nvidia-smi,我们能看到Driver Version 456.43,所以我最高能装到CUDA 11.1 GA版本,所以CUDA11.0和10.1都能装。
在这里插入图片描述
在这里插入图片描述
方法二:鼠标右键点击桌面,选择NVIDIA 控制面板,在主页也能看到版本信息,然后也点击系统信息进一步查看
在这里插入图片描述
在这里插入图片描述

在这里插入图片描述
这也能看到CUDA的版本信息

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/一键难忘520/article/detail/1013474
推荐阅读
相关标签