您的位置:首页 > 其它

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

2018-03-23 18:44 1786 查看
今天在linux系统,gtx 960显卡, cuda 9.1版本上用pytorch操作显卡,报出了如下warning“PyTorch no longer supports this GPU because it is too old.”,且程序出现 Runtime Error。谷歌半天后,解决方法如下:
首先,我的 pytorch 是在安装完 anaconda 后用 pytorch 官网给出的 conda 指令直接安装的,谷歌上的大牛说这种方法对于一些老版本gpu不行,得卸载 pytorch 改用 source 的方法重新安装。大牛的具体说法如下: https://discuss.pytorch.org/t/found-gpu0-geforce-gtx-960m-which-is-of-cuda-capability-5-0/15135 卸载 pytorch 的指令:
conda uninstall pytorch

根据里面的提示,我们参考下面这个网站进行 pytorch 的 source 安装: https://github.com/pytorch/pytorch#installation 注意,当这里面的指令涉及到 cuda 版本的时候要根据自己电脑实际安装的 cuda 版本来填写,例如,cuda 8.0 就在指令中写 cuda80,我安装的是 cuda 9.1 所以在指令中写的就是 cuda91。另外就是,教程中需要填写 anaconda 的安装目录,不知道的小伙伴只要在终端执行“which anaconda”,就能得到它的完整路径了。

之后,根据里面的教程很顺利的安装完 pytorch 了,然后就迫不及待地打开 python 然后 import torch,结果又报错,can not import name_c,然后又查查查,发现问题居然是不能在 pytorch 的根目录 import torch,重新打开一个终端,再 import 一次问题就自动消失了(瀑布汗......)。
大牛的分析在此: https://blog.csdn.net/draco_mystack/article/details/71191924
这样再用 pytorch 操作显卡虽然仍会抛出 warning,但是不会影响程序运行。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐