Ubuntu20.04下Pycharm运行aihwkit示例报错找不到aihwkit.inference模块
aihwkit.inference模块导入报错解决方案
方案1:脚本内临时添加导入路径
在examples/06_lenet5_hardware_aware.py文件的所有导入语句前,添加以下代码:
import sys sys.path.append("/home/你的用户名/自定义路径/aihwkit/src")
请将路径替换为你本地实际的src文件夹绝对路径
运行脚本时Python会优先从你指定的路径查找模块,即可正常导入aihwkit.inference相关内容。
方案2:Pycharm中配置源根目录(永久生效)
- 在Pycharm中打开aihwkit源码项目
- 找到项目根目录下的
src文件夹,右键点击 - 选择
Mark Directory as→Sources Root - 重启Pycharm后重新运行示例即可
方案3:从源码安装开发版aihwkit
你出现报错的核心原因大概率是pip默认安装的稳定版aihwkit未包含inference模块,可直接从本地源码安装最新版本:
打开终端切换到aihwkit源码的根目录,运行以下命令:
pip install -e .
该命令会以可编辑模式安装当前源码包,后续所有源码修改会实时生效,也能正常识别所有新增模块。
额外排查点
检查src/aihwkit/inference目录下是否存在__init__.py文件,如果缺失新建一个空的该文件即可,Python需要该文件将目录识别为可导入模块。
内容的提问来源于stack exchange,提问作者user14092875
相关产品推荐
相关产品推荐

