Skip to content

Instantly share code, notes, and snippets.

View nexgus's full-sized avatar

Augustus Chen nexgus

  • New Taipei City, Taiwan
  • 03:55 (UTC +08:00)
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__)'