ImportError: libc10.so: cannot open shared object file: No such file or directory

解决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

你可能感兴趣的:(python基础,python)