How to use mkldnn on conda installation?

1

If I use conda tool, that is, “conda install mkldnn” to install mkldnn, then I don`t know what to do in the next step for the link between mkldnn and pytorch, please give me some instructions if possible?Thanks a lot!

2

I guess you should first inspect if a pre-compiled pytorch is already using mkldnn:

import torch
print(*torch.__config__.show().split("\n"),sep="\n")

If USE_MKLDNN=ON, then you are already using it. Otherwise, the other simplest way I have in mind if you want to bind Pytorch and mkldnn is to build it from source.

3

Thanks a lot, I will try it to make sure that the mkldnn is working!