新电脑重新安装pytorch,记录从头配置的过程


目录

  • 安装对应版本的CUDA toolkit
  • 查看对应版本
  • 下载对应版本
  • toolkit安装
  • 命令行运行是否安装成功
  • 安装CudNN
  • 下载对应版本
  • 安装
  • 添加系统环境路径
  • 验证是否安装成功


安装对应版本的CUDA toolkit

查看对应版本

电脑打开NVIDIA控制面板

gpustat 配置环境变量 windows配置cuda环境变量_windows


选择“帮助”中的“系统信息”

gpustat 配置环境变量 windows配置cuda环境变量_系统环境_02


控制面板看版本号

gpustat 配置环境变量 windows配置cuda环境变量_系统环境_03

通过“组件”查看对应安装cuda的版本,个人电脑是11.1

gpustat 配置环境变量 windows配置cuda环境变量_系统环境_04

下载对应版本

cuda toolkit 链接 在列表中选择和自己电脑对应的cuda版本。进入下载页面

gpustat 配置环境变量 windows配置cuda环境变量_深度学习_05

在下载页面中,确定相应配置,windows 10,选择“exe(local)",进行下载

gpustat 配置环境变量 windows配置cuda环境变量_windows_06

toolkit安装

双击exe进行安装,设置安装路径

进入安装页面,可以直接”精简模式“快速安装

gpustat 配置环境变量 windows配置cuda环境变量_解压缩_07


这里我选择自定义为了设置安装路径中间出现了因为visual studio 版本不匹配的问题,重新去下载了vs2019

gpustat 配置环境变量 windows配置cuda环境变量_解压缩_08


配置好以后开始安装

gpustat 配置环境变量 windows配置cuda环境变量_深度学习_09


安装完成

gpustat 配置环境变量 windows配置cuda环境变量_系统环境_10


检查系统环境变量是否导入了cuda

gpustat 配置环境变量 windows配置cuda环境变量_gpustat 配置环境变量_11

命令行运行是否安装成功

nvcc --version
set cuda

gpustat 配置环境变量 windows配置cuda环境变量_解压缩_12


完成!成功!

安装CudNN

下载对应版本

官网下载链接 进入NVIDIA主页,选择platforms,选择cudNN

gpustat 配置环境变量 windows配置cuda环境变量_深度学习_13


选择download,需要先注册一下+填写问卷

都完成了以后就能进入下载页面了

gpustat 配置环境变量 windows配置cuda环境变量_windows_14


如果没有符合自己cuda版本的cudnn,可以选择Archived releases 查看更多版本

这里我选择了下面的版本

gpustat 配置环境变量 windows配置cuda环境变量_windows_15

安装

下载完成以后是一个压缩包,解压缩

gpustat 配置环境变量 windows配置cuda环境变量_深度学习_16


将解压缩后的三个文件夹拷贝到cuda配置的路径中,就是set cuda 命令后展示的路径.两个文件夹中有同名的文件夹也没关系。直接复制即可。

gpustat 配置环境变量 windows配置cuda环境变量_解压缩_17

添加系统环境路径

往系统环境变量中的 path 添加如下路径(根据自己的路径进行修改)
{cuda 安装路径} \bin
{cuda 安装路径} \include
{cuda 安装路径} \lib
{cuda 安装路径} \libnvvp

验证是否安装成功

首先win+R启动cmd,cd到安装目录下的 …\extras\demo_suite,然后分别执行bandwidthTest.exe和deviceQuery.exe

gpustat 配置环境变量 windows配置cuda环境变量_gpustat 配置环境变量_18


gpustat 配置环境变量 windows配置cuda环境变量_gpustat 配置环境变量_19


都得到pass,配置成功!