Solve the problem that pytorch does not support GTX9 graphic cards (pytorch新版本不支持GTX9系显卡的问题与解决方案)

在更新系统后安装PyTorch时遇到GPU不支持的问题,由于CUDA能力过旧,PyTorch不再支持GTX9系列显卡。通过源码安装PyTorch可以解决此问题,但需要注意torchvision的安装。直接使用pip安装torchvision会触发PyTorch最新版的自动安装,导致问题循环。避免使用Anaconda,因为它会自动安装最新版PyTorch。可行的解决方案是从torchvision源代码安装,规避pip库中依赖文件的缺失问题。虽然仍可能收到不支持显卡的警告,但实际上程序已使用GPU运行。
摘要由CSDN通过智能技术生成

最近在更新了系统后安装pytorch,发现会出现显卡太老已经不支持的问题:

Found GPU0 GeForce GTX 960M which is of cuda capability 5.0.

    PyTorch no longer supports this GPU because it is too old.

讲道理,我的960M也不算太渣的显卡吧,毕竟学生党买不起泰坦。在google上搜索后,开发者说因为新版本的性能需求等原因,最新的pytorch都不再支持9系显卡及其以下的了。建议的解决方案是:从源进行安装 install from source.

按照以下的方法进行源码安装pytorch(坑并不在这里,这里基本没有问题)

git clone --recursive https://github.com/pytorch/pytorch
cd pytorch
python setup.py install

如果显示缺少什么工具,就按照提示来一步步安装

真正的坑在于torchvision

如果你按照官网的指令,只用pip安装的话,你会发现错误:torchvision已经不在torch的官网库里。。就是类似的错误吧,它建议你使用annaconda进行安装!

但是请不要去用annaconda安装!!!因为annaconda的服务太好了(毕竟这个工具的特点就是能自动帮你解决包之间的依赖问题),所以即使你只用annaconda安装torchvision,它也会同时帮助你安装pytorch的最新版本ÿ

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值