使用CMake打包并安装Python绑定的问题求助
.so库的问题 我之前也踩过这个一模一样的坑!当用SWIG生成Python绑定后,通过CMake安装出现foo has no attribute bar(),本质就是Python导入foo模块时,找不到对应的底层共享库_foo.so——毕竟只有当两者在同一目录时,Python的模块搜索机制才能自动定位到它。下面是几个我亲测有效的解决方案:
方案1:调整CMake安装路径,让.py和.so同目录
这是最直接的根治办法,把SWIG生成的foo.py和编译出来的_foo.so都安装到Python的site-packages目录下(Python默认的模块搜索路径之一)。在CMake里可以这么配置:
# 先找到Python3的开发依赖 find_package(Python3 REQUIRED COMPONENTS Interpreter Development) # 生成SWIG绑定库 swig_add_library(foo LANGUAGE python SOURCES foo.i) swig_link_libraries(foo Python3::Python your_cpp_target_name) # 替换成你的C++目标库名称 # 安装foo.py到Python的site-packages目录 install(FILES ${CMAKE_CURRENT_BINARY_DIR}/foo.py DESTINATION ${Python3_SITEPACKAGES}) # 安装_foo.so库到同一个site-packages目录 install(TARGETS foo LIBRARY DESTINATION ${Python3_SITEPACKAGES})
安装完成后,foo.py和_foo.so会共存于Python的标准模块目录,导入时自然能找到依赖。
方案2:修改生成的foo.py,添加库搜索路径
如果因为项目结构限制,不想把.so放到site-packages,可以在SWIG生成的foo.py开头添加代码,把.so所在的目录加入Python的搜索路径。
比如你的_foo.so安装在${CMAKE_INSTALL_PREFIX}/lib,而foo.py在${CMAKE_INSTALL_PREFIX}/python,可以在CMake里动态修改foo.py:
# 先将文件安装到目标路径 install(FILES ${CMAKE_CURRENT_BINARY_DIR}/foo.py DESTINATION ${CMAKE_INSTALL_PREFIX}/python) install(TARGETS foo LIBRARY DESTINATION ${CMAKE_INSTALL_PREFIX}/lib) # 给foo.py添加路径配置代码 file(APPEND ${CMAKE_CURRENT_BINARY_DIR}/foo.py " import sys import os # 从foo.py的位置向上定位到lib目录 lib_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), '../lib') sys.path.append(os.path.abspath(lib_path)) ")
这样当Python导入foo时,会自动把lib目录加入搜索路径,就能找到_foo.so了。
方案3:通过环境变量PYTHONPATH指定库路径
这是临时生效的快速调试方案,适合测试阶段。如果_foo.so安装在/usr/local/lib/foo_lib,只需要在运行Python前设置环境变量:
export PYTHONPATH=/usr/local/lib/foo_lib:$PYTHONPATH
如果要永久生效,可以把这条命令添加到你的shell配置文件(比如~/.bashrc或~/.zshrc),或者系统级的环境变量配置文件中。
额外排查要点
- 安装后检查
foo.py和_foo.so的实际位置是否符合预期,文件权限是否为可读; - 确认
foo.py里的导入语句是import _foo(SWIG默认生成的是这个名字,不要手动修改); - 如果是Windows系统,对应的库文件是
_foo.pyd,排查逻辑和Linux一致,只是文件扩展名不同。
内容的提问来源于stack exchange,提问作者dirtshell

