pytorch错误

 

python: /pytorch/third_party/ideep/mkl-dnn/src/cpu/jit_avx2_conv_kernel_f32.cpp:582: static mkldnn::impl::status_t mkldnn::impl::cpu::jit_avx2_conv_fwd_kernel_f32::init_conf(mkldnn::impl::cpu::jit_conv_conf_t&, const convolution_desc_t&, const mkldnn::impl::memory_desc_wrapper&, const mkldnn::impl::memory_desc_wrapper&, const mkldnn::impl::memory_desc_wrapper&, const primitive_attr_t&, bool, float): Assertion `jcp.ur_w * (jcp.nb_oc_blocking + 1) <= num_avail_regs' failed.
Aborted

参考：https://github.com/pytorch/pytorch/issues/20583

 

解决办法：

原来使用的torch版本是1.0.1.post2

使用命令：

pip install --pre torch -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html

更换成1.5.0版本即可，就能成功运行了