admin 管理员组

文章数量: 1184232

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.is_available(

本文标签： 显卡 流程 版本 CUDA Windows

版权声明：本文标题：Windows下显卡支持的cuda及配套GPU版本pytorch下载全流程（2022.03.27） 内容由网友自发贡献，该文观点仅代表作者本人， 转载请联系作者并注明出处：http://www.roclinux.cn/b/1767828945a3508962.html， 本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容，一经查实，本站将立刻删除。