在使用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版本了

Logo

更多推荐