安装好pytorch后 import torch报错
出现错误:
File "/home1/wang-internship-1/anaconda3/lib/python3.7/site-packages/torch/__init__.py", line 81, in <module>
from torch._C import *
ImportError: /home1/wang-internship-1/anaconda3/lib/python3.7/site-packages/torch/lib/../../../../libmkl_gnu_thread.so: undefined symbol: mkl_blas_zgemm_blk_info_hi_thr_bdz
报错原因
cuda版本与pytorch版本不匹配
解决办法
更换cuda版本,我原本是cuda10.1版本,后来卸载后跟换成10.0版本后,问题解决了。
conda uninstall cudatoolkit conda install cudatoolkit=10.0