import torch報錯


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

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM