发布于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版本
- python
- >>> import torch
- >>> torch.version.cuda
- '10.0.130'
查看机器的cuda版本
- nvcc -V
- ...
- 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
作者:骏马
链接:https://www.pythonheidong.com/blog/article/235007/2801bce3075e27d7c90f/
来源:python黑洞网
任何形式的转载都请注明出处,如有侵权 一经发现 必将追究其法律责任
昵称:
评论内容:(最多支持255个字符)
---无人问津也好,技不如人也罢,你都要试着安静下来,去做自己该做的事,而不是让内心的烦躁、焦虑,坏掉你本来就不多的热情和定力
Copyright © 2018-2021 python黑洞网 All Rights Reserved 版权所有,并保留所有权利。 京ICP备18063182号-1
投诉与举报,广告合作请联系vgs_info@163.com或QQ3083709327
免责声明:网站文章均由用户上传,仅供读者学习交流使用,禁止用做商业用途。若文章涉及色情,反动,侵权等违法信息,请向我们举报,一经核实我们会立即删除!