关于50系显卡pytorch不兼容问题解决办法
在使用50系显卡训练大模型时出现以下问题我这里选择的是cuda13.0版本下载的,因为我的cuda版本是13.0,在终端输入nvidia -smi可查看你的cuda版本,选择对应的即可。
·
在使用50系显卡训练大模型时出现以下问题
这个是指当前50系显卡计算能力sm_120与当前pytorch版本不兼容,所以需要更新pytorch,以下是解决步骤:
一:访问pytorch官方网站,根据需求选择对应的pytorch版本:
网站如下:开始使用

我这里选择的是cuda13.0版本下载的,因为我的cuda版本是13.0,在终端输入nvidia -smi可查看你的cuda版本,选择对应的即可
二:在终端执行由官网提供的下载网址:
pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu130
注意:这里还需要查看当前的python版本,如果python版本太低的情况下在终端执行pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu130会出现以下问题:

这里只需要更新python版本就行了,我这里直接把python更新到最新版本3.13.5了。
更新python版本后再在终端执行pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu130就可以下载选定的pytorch版本了
更多推荐


所有评论(0)