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

今天在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,但是不会影响程序运行。

  • 1
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值