记录深度学习GPU配置,下载CUDA与cuDnn,安装tensorflow

目标下载:

cuda 11.0.1_451.22 win10.exe
cudnn-11.0-windows-x64-v8.0.2.39.zip

Anaconda的安装请看别的博主的,这里不再赘述

看看自己电脑的cuda

方法一:打开英伟达面板查看

方法二:使用命令行

随便找个文件夹,在顶部路径输入"cmd"

输入下面命令

nvidia-smi

我的目标下载cuda11.0,由于11.0<12.0,可以

cuda历史版本网址

CUDA Toolkit Archive | NVIDIA Developer

自己下载过11.0.1版本

点击下载local版本,本地安装,有2个多GB,很大,我不喜欢network版本,容易掉线

cuDnn

https://developer.nvidia.com/cudnn

https://developer.nvidia.com/cudnn-downloads

cuDNN Archive | NVIDIA Developer

上面使用qq邮箱未必成功,我自己是用谷歌邮箱注册的

如何安装cuda,你可以看别的博主的文章,我不再赘述

TensorFlow问题

TensorFlow的安装或是pytorch的安装一定看你的版本,还有,

我不推荐在线下载,推荐本地安装,!!

我不推荐在线下载,推荐本地安装,!!

我不推荐在线下载,推荐本地安装,!!

TensorFlow先看cuDa与cuDnn版本,比如我的cuDnn与cuda分别是8.0.2与11.0,对应TensorFlow2.4.0,
 

TensorFlow官网版本查看: https://tensorflow.google.cn/install/source_windows?hl=en#gpu

目标下载2.4.0

要下载包,找Pypi,Pypi上除了pytorch,其他的任何第三方python包你都可以找到

PyPi网站:

https://pypi.org/search/?q=tensorflow-gpu&o=

我的python是3.6,所以下载3.6的

如何看看自己python版本

...额,先进入Anaconda Prompt,...这个留着说

TensorFlow的包与Keras包有版本捆绑问题,不对应版本的话运行会出错

这里TensorFlow的包与Keras包版本参见另外一篇博主文章

tensorflow与keras版本对应_pthon3.9.7 应该安装什么版本的tensorflow-CSDN博客

https://pypi.org/search/?q=keras

依旧在pypi搜索keras

区别在于,tar.gz是压缩包,解压后会在你的C盘留有无用缓存,记得定时清理

whl版本直接安装没有缓存