最近在更新了系统后安装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的最新版本ÿ