解决lib/python3.7/site-packages/torch/lib/libtorch_cpu.so: undefined symbol: iJIT_NotifyEvent

问题描述

在创建新的虚拟环境并安装相应包后出现：
import torch
File “/mnt/sda/home/cxz/.conda/envs/bat/lib/python3.7/site-packages/torch/init.py”, line 218, in
from torch._C import * # noqa: F403
ImportError: /mnt/sda/home/cxz/.conda/envs/bat/lib/python3.7/site-packages/torch/lib/libtorch_cpu.so: undefined symbol: iJIT_NotifyEvent

问题原因

mkl包版本不匹配
PyTorch 的 CPU 库是在旧版本的 MKL 上构建的，而新版（例如 MKL 2024.1）中移除了一些符号，比如 iJIT_NotifyEvent。这就会导致在运行时找不到该符号，从而产生错误。

解决方法

参考链接：https://blog.csdn.net/mr_hore/article/details/138961434
对mkl进行降级

conda install mkl=2024.0