Is this possible somehow? Or at least to disable MKL-DNN on runtime?
I found this for runtime:
torch.backends.mkldnn.flags(enabled=False)
but I get the following error:
Traceback (most recent call last):
File “”, line 1, in
AttributeError: module ‘torch.backends’ has no attribute ‘mkldnn’