本文主要是介绍记录深度学习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版本直接安装没有缓存
这篇关于记录深度学习GPU配置,下载CUDA与cuDnn,安装tensorflow的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!