admin 管理员组文章数量: 887021
新电脑重新安装pytorch,记录从头配置的过程
参考博文:https://blog.csdn/weixin_43848614/article/details/117221384
目录
- 安装对应版本的CUDA toolkit
- 查看对应版本
- 下载对应版本
- toolkit安装
- 命令行运行是否安装成功
- 安装CudNN
- 下载对应版本
- 安装
- 添加系统环境路径
- 验证是否安装成功
安装对应版本的CUDA toolkit
查看对应版本
电脑打开NVIDIA控制面板
选择“帮助”中的“系统信息”
控制面板看版本号
通过“组件”查看对应安装cuda的版本,个人电脑是11.1
下载对应版本
cuda toolkit 链接
在列表中选择和自己电脑对应的cuda版本。进入下载页面
在下载页面中,确定相应配置,windows 10,选择“exe(local)",进行下载
toolkit安装
双击exe进行安装,设置安装路径
进入安装页面,可以直接”精简模式“快速安装
这里我选择自定义为了设置安装路径
中间出现了因为visual studio 版本不匹配的问题,重新去下载了vs2019
配置好以后开始安装
安装完成
检查系统环境变量是否导入了cuda
命令行运行是否安装成功
nvcc --version
set cuda
完成!成功!
安装CudNN
下载对应版本
官网下载链接
进入NVIDIA主页,选择platforms,选择cudNN
选择download,需要先注册一下+填写问卷
都完成了以后就能进入下载页面了
如果没有符合自己cuda版本的cudnn,可以选择Archived releases 查看更多版本
这里我选择了下面的版本
安装
下载完成以后是一个压缩包,解压缩
将解压缩后的三个文件夹拷贝到cuda配置的路径中,就是set cuda 命令后展示的路径.两个文件夹中有同名的文件夹也没关系。直接复制即可。
添加系统环境路径
往系统环境变量中的 path 添加如下路径(根据自己的路径进行修改)
{cuda 安装路径} \bin
{cuda 安装路径} \include
{cuda 安装路径} \lib
{cuda 安装路径} \libnvvp
验证是否安装成功
首先win+R启动cmd,cd到安装目录下的 …\extras\demo_suite,然后分别执行bandwidthTest.exe和deviceQuery.exe
都得到pass,配置成功!
版权声明:本文标题:windows环境下CUDA安装及配置 内容由网友自发贡献,该文观点仅代表作者本人, 转载请联系作者并注明出处:http://www.freenas.com.cn/jishu/1716032214h663195.html, 本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容,一经查实,本站将立刻删除。
发表评论