程序员最近都爱上了这个网站  程序员们快来瞅瞅吧!  it98k网:it98k.com

本站消息

站长简介/公众号

  出租广告位,需要合作请联系站长

+关注
已关注

分类  

暂无分类

标签  

暂无标签

日期归档  

2023-05(5)

2023-06(1)

undefined symbol: _ZN2at19UndefinedTensorImpl10_singletonE

发布于2020-02-27 10:46     阅读(3688)     评论(0)     点赞(19)     收藏(5)


复现Graph R-CNN for Scene Graph Generation这篇论文的代码时报错。

在执行训练命令时,报错ImportError: /.../_C.cpython-36m-x86_64-linux-gnu.so: undefined symbol: _ZN2at19UndefinedTensorImpl10_singletonE

原因:cuda、torch、torchvision版本不对应。

 查看torch的cuda版本

  1. python
  2. >>> import torch
  3. >>> torch.version.cuda
  4. '10.0.130'

查看机器的cuda版本

  1. nvcc -V
  2. ...
  3. Cuda compilation tools, release 9.0, V9.0.176

发现并不对应,去官网找到符合自己要求的版本https://pytorch.org/get-started/previous-versions/

论文要求是torch=1.0、cuda=8.0+

conda install pytorch==1.0.0 torchvision==0.2.1 cuda90 -c pytorch

解决问题!虽然报了新的错误,但是至少这个错误解决了,历时两天:)

参考博客:https://blog.csdn.net/qq1483661204/article/details/103834224

 

发布了20 篇原创文章 · 获赞 9 · 访问量 3万+


所属网站分类: 技术文章 > 博客

作者:骏马

链接:https://www.pythonheidong.com/blog/article/235007/2801bce3075e27d7c90f/

来源:python黑洞网

任何形式的转载都请注明出处,如有侵权 一经发现 必将追究其法律责任

19 0
收藏该文
已收藏

评论内容:(最多支持255个字符)