Mkl-dnn problem with pytorch. I run on a university cluster, so not able to change the mkl-dnn version

python3: /pytorch/third_party/ideep/mkl-dnn/src/common/scratchpad.cpp:63: mkldnn::impl::global_scratchpad_t::global_scratchpad_t(size_t): Assertion `scratchpad_ != nullptr’ failed.
./setup_venv_on_cluster.sh: line 21: 19312 Aborted (core dumped)