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

本文详细介绍了在Windows上为MX150显卡更新驱动、下载安装CUDA Toolkit 10.1及对应GPU版本PyTorch的步骤,并提供了验证安装成功的教程。
摘要由CSDN通过智能技术生成

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

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

1、打开英伟达驱动下载官网:https://www.nvidia.cn/geforce/drivers/
2、直接下载自动更新程序,安装流程很简单。驱动下载官网
3、打开下好的驱动更新软件,选择自动更新驱动,
NVIDIA驱动更新软件
4.打开NVIDIA控制面板——系统信息——组件——可以看到我mx150现在支持的是11.6.127版本CUDA。
查看支持的CUDA版本

二、官网下载相应版本CUDA

1、打开网址https://developer.nvidia.com/cuda-toolkit-archive
2、由于我之前没有更新驱动,我下载的是CUDA Toolkit 10.1 update2,选择local,可以用迅雷接管下载,速度比较快。
CUDA Toolkit 10.1 update23、打开安装程序,可以选择默认路径安装在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
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值