解决5070显卡出现pytorch不支持sm_120问题

1.当前50系显卡不支持现有版本的pytorch,会出现如下报错:

2.解决办法
(1)打开pytorch官网:Get Started

(2)选择Preview (Nightly)在选择Cuda 12.8,在Run this Command:命令中复制链接网址:

https://download.pytorch.org/whl/nightly/cu128

(3)在google浏览器(建议,其他浏览器可能出现打不开的情况)中打开上述复制的网址:

https://download.pytorch.org/whl/nightly/cu128

(4)选择torch,根据自己电脑的配置和系统,选择对应的cpu或者gpu版本(建议选择最新日期的)

(5)返回首页后点击torchvison,选择对应的torchvison版本(如果需要的话)

(5)注意选择的torchvison版本日期需要比torch的版本晚一天!!
如选择torch:torch==2.7.0.dev20250309+cu128
则需要对应选择torchvison:torchvision-0.22.0.dev20250310+cu128
3.Anaconda环境创建
(1)创建对应的python版本,如torch文件为:torch-2.8.0.dev20250401+cu128-cp310-cp310-win_amd64则需创建python==3.10,激活环境之后,运行如下命令:
pip install "下载文件的保存路径",依次安装torch和torchvison


(2)torch环境已经安装完成,再安装所需要的包即可
4.环境检验
(1)导入conda 环境之后,编写如下python文件:

(2)运行结果出现:“sm_120”和True则说明环境安装成功!

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值