admin管理员组

文章数量:1566332

Windows下显卡支持的cuda及配套GPU版本pytorch下载全流程(2022.03.27)

一、更新电脑显卡驱动(以mx150为例)

1、打开英伟达驱动下载官网:https://www.nvidia/geforce/drivers/
2、直接下载自动更新程序,安装流程很简单。
3、打开下好的驱动更新软件,选择自动更新驱动,

4.打开NVIDIA控制面板——系统信息——组件——可以看到我mx150现在支持的是11.6.127版本CUDA。

二、官网下载相应版本CUDA

1、打开网址https://developer.nvidia/cuda-toolkit-archive
2、由于我之前没有更新驱动,我下载的是CUDA Toolkit 10.1 update2,选择local,可以用迅雷接管下载,速度比较快。
3、打开安装程序,可以选择默认路径安装在c盘,选择自定义安装,如果电脑没有安装VS 的话需要取消vs前的勾,还有把每个小加号点开,如果新版本的版本号比你当前版本的版本号还要低的选项也要取消,之后点击继续。

4.下面就可以验证cuda是否安装成功:
在运行里面输入“cmd” 然后用命令“nvcc -V”进行测试
即如下的代码,此处需要注意的是:中间是有空格的,之后按回车键。

nvcc -V

运行结果如下,可以看到安装成功v10.1.243.

三、安装对应版本torch

1、如果已经下过torch可以先验证一下行不行
运行以下代码,如果返回true证明当前torch版本与cuda是对应的且可以调用。若返回false,则进行下一步。

import torch
print(torch.cuda

本文标签: 显卡流程版本CUDAWindows