CUDA的安装

我的公众号:硕哥聊AI,关注直接下载代码

我的抖音号:硕哥讲AI

我的微信:18810952966,加我入群AI同行共同促进AI知识传播
cuda 安装tensorflow-gpu,cudnn
tensorflow是什么,从哪里来,做什么的问题,这节课呢给大家讲解一下cudnn安装的问题,为什么要先讲这个呢,我们知道有这么一句话,工欲善其事必先利其器,因为安装了cudnn之后呢,在配合tensorflow-Gpu能够使你的程序运算速度增加很多倍。
好,按照我们一贯的学习思路先把思维导出画出来,
第一章cuda是什么cudnn是什么,为什么要安装cuda,
第二章是查看显卡信息,
第三章是安装cuda,
第四章是安装cudnn,
第五章安装tensorflow-gpu,最后测试运行。

第一章:cuda是什么cudnn是什么?

CUDA(Compute Unified Device Architecture), 是显卡厂家NVIDIA推出的运算平台,CUDA是一种并行计算架构,此架构使用GPU解决复杂的计算问题。
1.CUDNN是一个常见的神经网络层加速库文件,能够很大程度把加载到显卡上的网络层数据进行优化计算。 而CUDA就像一个很粗重的加速库,其主要依靠的是显卡。
2.CUDNN需要在有CUDA的基础上进行。
3.CUDNN可以在CUDA基础上加速2倍以上。这些都是我试验出来的

第二章查看显卡信息:

查看自己电脑显卡,在桌面上鼠标右键点击,点击NVIDIA控制面板,点击帮助(H),点击系统信息,点击组件,看到NVCUDA.DLL是10.1.120,这个就是我们显卡对应的cuda版本。
cuda 安装tensorflow-gpu,cudnn

第三章:安装cuda ,cudnn

地址 : https://developer.nvidia.com/cuda-toolkit-archive.
下面是各种版本,我们选择10.1版本就行了,因为电脑是10.1.120
cuda 安装tensorflow-gpu,cudnn
我们点击版本出现如下界面
cuda 安装tensorflow-gpu,cudnn
选择windows,x86_64,10,exe(local),点击download,下载完之后,正常安装即可。安装完成之后,环境变量自动配置,如果没有配置上可以手动配置,环境变量是什么?这里不解释,可以自行度娘。
cuda 安装tensorflow-gpu,cudnn
在控制台即可查看是否安装成功nvcc -V
cuda 安装tensorflow-gpu,cudnn
然后是安装cudnn,地址,链接: https://developer.nvidia.com/rdp/cudnn-archive
cuda 安装tensorflow-gpu,cudnn
然后选择,解压之后把里面对应的文件拷贝到cuda安装目录,如下图所示
cuda 安装tensorflow-gpu,cudnn

第四章安装tensorflow-Gpu

安装tensorflow-GPU也是根据版本来的,参考链接。
链接: https://tensorflow.google.cn/install/source_windows.
cuda 安装tensorflow-gpu,cudnn
然后就可以根据,版本安装了,安装的时候添加镜像会比较快,安装的命令如下pip install tensorflow-gpu 1.13 -i http://pypi.douban.com/simple --trusted-host pypi.douban.com然后就是测试,测试代码,我会同步更新到我的公众号,大家可以直接下载,这个时候我们测试gpu利用率,命令行nvidia-smi -l 1。
好了以上就是今天的内容,最后送给大家一句话,如果安装不成功直接安装cpu 的版本也可以
pip install tensorflow-cpu 1.13 -i http://pypi.douban.com/simple --trusted-host pypi.douban.com

相关文章:

  • 2021-04-23
  • 2021-08-04
  • 2021-04-24
  • 2022-12-23
  • 2021-05-08
  • 2021-12-04
  • 2021-09-07
  • 2021-08-25
猜你喜欢
  • 2021-09-30
  • 2021-12-20
  • 2021-12-22
  • 2021-06-01
  • 2021-12-22
  • 2021-08-23
  • 2021-05-10
相关资源
相似解决方案