报错 

在尝试导入 PyTorch 时网络训练报错ImportError: /opt/conda/envs/edm/lib/python3.9/site-packages/torch/lib/libtorch_cpu.so: undefined symbol: iJIT_NotifyEvent 

 遇到未定义的iJIT_NotifyEvent 符号报错

解决办法

该错误原因是 MKL(Math Kernel Library)版本与 PyTorch 版本不兼容。iJIT_NotifyEvent 符号在 MKL 2024.1 版本中被移除,而 PyTorch 是根据旧版本的 MKL 构建的。MKL 全称 Intel Math Kernel Library, 是由 Intel 公司开发的,专门用于矩阵计算的库。

将 MKL 降级到 2024.0 版本,即可解决此问题。 

conda install mkl=2024.0

Logo

北京人形旗下天工造物具身智能开源社区,聚焦具身天工与慧思开物两大平台

更多推荐