Google Colab导入.pyx文件报ModuleNotFoundError问题求助

.pyx为Cython源码文件，无法直接被Python解释器识别导入，必须先编译为对应系统的扩展模块（Linux下为.so文件）后才可正常导入，你遇到的报错大概率是未完成编译步骤导致的。

步骤1：先排查路径配置问题 #

先确认路径和文件真实存在，执行如下命令验证：

!ls /content/drive/My Drive/desiredPath

如果输出结果中没有你的desiredModule.pyx文件，说明你添加的sys.path路径有误，可执行!find /content/drive -name "desiredModule.pyx"查找文件真实路径后修正sys.path.append的参数。注意Colab挂载的Drive路径部分版本显示为/content/drive/MyDrive（无空格），需和实际路径保持一致。

步骤2：安装Cython依赖 #

先安装编译所需的基础依赖：

!pip install cython
!apt-get install -y build-essential

步骤3：编译并导入模块 #

两种常用编译方案可选：

方案A：使用pyximport自动编译（适合无复杂C依赖的简单场景） #

在你现有代码的导入模块步骤前，添加pyximport初始化代码即可：

from google.colab import drive
drive.mount('/content/drive')
import sys
sys.path.append('/content/drive/My Drive/desiredPath')

# 新增以下两行
import pyximport
pyximport.install()

from desiredModule import desiredClass

该方案会在导入时自动完成编译操作，无需额外配置。

方案B：手动编译为扩展模块（适合有C依赖的复杂场景） #

1. 在你的desiredPath目录下新建setup.py文件，写入如下内容：

from setuptools import setup
from Cython.Build import cythonize
# 若用到numpy等C扩展库，需额外导入并配置include路径
# import numpy

setup(
    ext_modules = cythonize("desiredModule.pyx"),
    # 有numpy依赖时新增下方参数
    # include_dirs = [numpy.get_include()]
)

2. 在Colab中执行编译命令：

!cd /content/drive/My Drive/desiredPath && python setup.py build_ext --inplace

3. 编译完成后，对应目录会生成.so格式的扩展模块，直接执行原导入语句即可：

from desiredModule import desiredClass

注意事项 #

• 每次修改.pyx源码后，需要重新执行编译操作才可生效
• 若编译报错提示缺少对应C库，可通过!apt-get install安装对应依赖后重新编译

内容的提问来源于stack exchange，提问作者BobelLegend