Skip to content

Instantly share code, notes, and snippets.

View abhayraw1's full-sized avatar

Abhay Rawat abhayraw1

View GitHub Profile
@mingfeima
mingfeima / pytorch_check_mkl_mkldnn.md
Last active July 8, 2022 06:09
BKMs to check whether mkl or mkldnn is enabled on PyTorch

BKMs to check whether mkl or mkldnn is enabled on PyTorch

PyTorch can be installed via different channels: conda, pip, docker, source code...

By default, mkl and mkl-dnn are enabled; But this might not always be true, so it is still useful to learn how to check this by yourself:

1. How to check whether mkl is enabled?

### check where your torch is installed
python -c 'import torch; print(torch.__path__)'