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

 

posted @ 2021-01-25 22:59  番茄--  阅读(1732)  评论(0)    收藏  举报