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