【PyTorch】libtorch_cpu.so: undefined symbol: iJIT_NotifyEvent
- 1 报错信息
- 2 原因
- 3 解决方法
1 报错信息
conda install pytorch=1.13.1 torchaudio=0.13.1 torchvision=0.14.1 cudatoolkit=11.7 -c pytorch
在 conda 环境中安装 torch 后测试，得到下面的错误：
>>> import torch
Traceback (most recent call last):File "<stdin>", line 1, in <module>File "/home/polaris/anaconda3/envs/xxx/lib/python3.9/site-packages/torch/__init__.py", line 218, in <module>from torch._C import * # noqa: F403
ImportError: /home/polaris/anaconda3/envs/xxx/lib/python3.9/site-packages/torch/lib/libtorch_cpu.so: undefined symbol: iJIT_NotifyEvent
2 原因
参考 issue: Link
这个符号在 MKL 2024.1 中被移除了，然而 PyTorch 是根据旧版本的 MKL 构建的。 MKL 全称 Intel Math Kernel Library， 是由 Intel 公司开发的，专门用于矩阵计算的库。
- 通过 conda 发布的 PyTorch 二进制文件是动态链接到 MKL 的，所以会遇到这个错误。
- 通过 pip 发布的 PyTorch 二进制文件是静态链接到 MKL 的（ 可以改用 pip install 来消除 MKL 的这个错误）
3 解决方法
参考 issue: Link
conda install mkl==2024.0
)
