编译pytorch RuntimeError: fft: ATen not compiled with MKL support解决方法
编译pytorch RuntimeError: fft: ATen not compiled with MKL support解决方法
1.安装mkl相关pkg
conda install mkl_fft mkl mkl-include
2.找到mkl安装的路径,对于我是在/opt/conda/lib和/opt/conda/include下.(主要看libmkl_intel_lp64.so,libmkl_gnu_thread.so,libmkl_core.so这几个库的位置)
3.修改pytorch/cm
原创
2020-12-15 15:57:17 ·
1905 阅读 ·
4 评论