Is pytorch lightning using MKLDNN when no GPU is available ? #5602
Answered
by
carmocca
AlexandreDey
asked this question in
Lightning Trainer API: Trainer, LightningModule, LightningDataModule
-
|
Hi, I was wondering if PL can use mkldnn/dnnl/onednn ? I may be mistaken, but it seems pytorch does not default to it and there is no explicit mention of it inside PL's code. If it's not, is there a way to enable it ? Alexandre |
Beta Was this translation helpful? Give feedback.
Answered by
carmocca
Apr 20, 2021
Replies: 1 comment
-
|
No, unfortunately, MKL-DNN is currently not supported Duplicate of #6020 |
Beta Was this translation helpful? Give feedback.
0 replies
Answer selected by
carmocca
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
No, unfortunately, MKL-DNN is currently not supported
Duplicate of #6020