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