解决ImportError: libc10.so: cannot open shared object file: No such file or directory
时间:2022年11月3日
背景:
在pytorch环境下,编写了一段cuda程序,并封装好,函数名叫做nms_cuda。想要调用nms_cuda,
from nms_cuda import nms
却报错:
Traceback (most recent call last):
File "/root/cuda_programming/nms/temp1.py", line 2, in
from nms_cuda import nms
ImportError: libc10.so: cannot open shared object file: No such file or directory
解决方法:
libc10.so是基于pytorch生成的,因此需要先导入torch包,然后再导入依赖于torch的包:
import torch
from nms_cuda import nms
编译成功。
————————————————
版权声明:本文为CSDN博主「beneficial」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/beneficial/article/details/127675796