安装好cuda和cudnn后,在虚拟环境下import torch报错:
from torch._C import * # noqa: F403
ImportError: /home/ccszbd/anaconda3/envs/esmfold_env/lib/python3.7/site-packages/torch/lib/libtorch_cpu.so: undefined symbol: iJIT_NotifyEvent
错误基本可以锁定的位置是:undefined symbol: iJIT_NotifyEvent。网上找了一圈,试过了各种方法,包括检查环境变量设置、检查cuda的版本与torch版本是否一致、torch为2.0以上的版本(我的版本是1.12)等等,各种方法都无法解决我的问题。最后,终于让我发现了华点~
解决方式
在安装torch时,可能由于mkl的版本与pytorch的版本不一致导致,或者缺少mkl依赖包,因此需要安装稳定的MKL依赖
pip install mkl==2024.0.0
需要注意的是,版本不一致可能是由于mkl的版本为2024.1导致的,因此需要降低mkl的版本,重新安装2024.0版本