Google Colab 看不到 .so 文件

Google Colab does not see .so files

本文关键字：文件 so 看不到 Colab Google      更新时间：2023-10-16

我正在尝试制作一个CUDA扩展，以便在Google Colab的Python中使用它。

这是 setup.py

sources = [os.path.join(path_to_cpp, 'interface.cpp'),
os.path.join(path_to_cpp, 'box_encoder_cuda.cu'),
os.path.join(path_to_cpp, 'random_horiz_flip.cu')]
setup(
name='SSD',
ext_modules=[
CUDAExtension(
'SSD._C', 
sources)
],
cmdclass={
'build_ext': BuildExtension
})

当我构建它时，我可以清楚地看到构建中存在 libc10.so

x86_64-linux-gnu-g++ -pthread -shared -Wl,-O1 -Wl,-Bsymbolic-functions -Wl,-Bsymbolic-functions -Wl,- 
z,relro -Wl,-Bsymbolic-functions -Wl,-z,relro -g -fstack-protector-strong -Wformat -Werror=format- 
security -Wdate-time -D_FORTIFY_SOURCE=2 build/temp.linux-x86_64-3.6/content/drive/My  
Drive/SSD/csrc/interface.o build/temp.linux-x86_64-3.6/content/drive/My Drive/SSD/csrc/box_encoder_cuda.o build/temp.linux-x86_64-3.6/content/drive/My  
Drive/SSD/csrc/random_horiz_flip.o -L/usr/local/lib/python3.6/dist-packages/torch/lib - 
L/usr/local/cuda/lib64 -lc10 -ltorch -ltorch_cpu -ltorch_python -lcudart -lc10_cuda -ltorch_cuda -o  
build/lib.linux-x86_64-3.6/SSD/_C.cpython-36m-x86_64-linux-gnu.so

但是，如果我尝试在 Colab 中使用它：

%%writefile test.py
import SSD._C
print("All ok")

!python test.py

我将看到以下错误消息：

Traceback (most recent call last):
File "test.py", line 1, in <module>
import SSD._C
ImportError: libc10.so: cannot open shared object file: No such file or directory

也许这是因为库没有在 Python 运行时中声明，我必须以某种方式声明它们？提前谢谢你！

附言 当我尝试直接在单元格中使用它而不写入.py文件时：

import SSD._C
print("All ok")

它产生

ModuleNotFoundError: No module named 'SSD'

在导入扩展之前导入火炬解决了问题：

%%writefile test.py
import torch
import SSD._C
print("All ok")

根据文档"这将解决动态链接器必须看到的一些符号"。

但是，为什么这个扩展名不能直接在.ipynb单元格中使用仍然是一个开放点。

编辑：如果在 .ipynb 中安装新扩展，请确保在使用前重新启动运行时。重新启动后，您将能够在 .ipynb 单元格中使用新扩展名

相关文章：