一、cuda及cudnn下载
1、查看自己电脑是否支持GPU
方法:鼠标移动到此电脑,点击鼠标右键,依次选择属性、设备管理器、显示适配器有以下图标(NVIDIA)即可安装GPU,我的是MX130的。
2、选择和自己电脑相匹配的cuda版本
2.1 查看自己电脑cuda版本:点击电脑左下角的搜索图标(放大镜)输入NVIDIA,选择如下应用
点击帮助、系统信息、组件找到NVCUDA64.DLL查看cuda版本,显示的是最高可安装的cuda版本,并非仅支持,我的是11.2.162版本
3、下载cuda
cuda官网:CUDA Toolkit Archive | NVIDIA Developer
游览器打开相比较于迅雷更慢,下载速度也很慢,建议直接复制网址进入迅雷搜索下载,进入后选择与自己电脑相匹配的版本,选择10.1
下载速度慢的另一个方法是通过像迅雷一样的第三方软件 Internet Download Manager下载(哔哩哔哩看到的,未亲测)
点击进入后,选择自己的系统,我的是WIN10系统(local完整版2.8G,network程序很小59.2MB),建议选择完整版,点击download下载完成即可
4、下载CUDNN 安装包
CUDNN官网:https://developer.nvidia.com/rdp/cudnn-download
方法和下载cuda一样,建议复制链接到迅雷下载
显示需要会员才能下载,点击立即加入,用邮箱注册一个账号,填写一个调查问卷即可下载
注册登陆后,勾选下列图标,进入CUDNN版本选择页
下图所显示的下面的链接表示10.2以前的版本,我的是11.2的,选择上面的版本,如下图
点开链接后选择和自己电脑系统匹配的CUDNN,点击进入,即跳转到下载界面,接着选择自己的下载路径即可
二、cuda安装
1、移动鼠标到cuda安装包上,选择以管理员身份运行,如下图
2、选择安装路径,选择不存在的,能够自己创建(建议选择默认路径,因为后面的安装需要相同的路径)
点击OK进入下一步
进入cuda安装环境后不要进行任何操作
返回电脑桌面,滑动鼠标右击此电脑—管理—服务和应用程序—服务
找到Windows Installer点击启动此服
选择自定义安装,下一步
下一步全部选择,电脑有更高的驱动的话,下面三个均可以不选
下一步选择安装位置(建议默认)
下一步如果有Visual Studio则勾选以下图标,没有则先安装Visual Studio最后,点击next进行安装,接下来就等着安装就可以了
我也是摸索了好几天,下了好几个版本,最终成功了cuda10.1
里面有cuda10.1+cudann,cuda10.0+cudann,cuda11.2+cudnn,有需要的自行下载
提取码:2458
tensorflow,torch详细配置见