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版本即可，就能成功运行了