安裝好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