解决ImportError: libc10.so找不到文件或目录的问题

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

编译成功。