在阅读完配置机器学习环境的首要知识之后就可以开始下载cuda和cudnn了
了解cuda、cudnn和tensorflow的版本要求,请转到博客:https://blog.csdn.net/tjb132/article/details/106911284
1,cuda下载
cuda官网下载:https://developer.nvidia.com/cuda-toolkit-archive新旧版本都有
我的是10.1的,直接点击 10.1 那个链接(有多个,选一个)。然后进入新界面之后按自己电脑选择就行。我的是windows,x64位。
查看gpu算力:
查看显卡版本和显卡对应算力水平请看官网:https://developer.nvidia.com/cuda-gpus
2,cudnn下载
cudnn的各个版本(包含最新)下载:(官网)[https://developer.nvidia.com/rdp/cudnn-download](https://developer.nvidia.com/rdp/cudnn-download)
记得勾选 I agree ,同意之后才会显示各个版本的cudnn
我的是windows版本,10.1的,所以如图点击这个就可以下载了
3,cuda 和 cudnn的安装教程+环境变量配置
具体的cuda 和 cudnn的安装教程网上很多。值得一提的是,一定要配置环境变量,这里特意只提供环境变量配置教程。
安装完cuda驱动之后,软件会自动添加部分环境变量到系统中,但是还要手动补充部分
这些是软件自动添加的
具体我就不说了,cuda的环境变量至少有4个,如图所示,看自己缺哪个就补哪个
注意,是在 path 中添加,不要删除原来的。
%cuda_path%\bin
%cuda_path%\lib\x64
%cuda_path%\extras\CUPTI\lib64
%cuda_path%\libnvvp