Aapo Kyrola f1fe4b1114 add simple memory analyzer and log warning if GPU underutilized (#21024) ...

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/21024

Add a new pybinded call to CUDAGetMemoryInfo.

Reviewed By: wesolwsk

Differential Revision: D15520607

fbshipit-source-id: f6d04e48f7d7cb089fc52fa8835cfee3f452d2f1

2019-05-28 19:58:54 -07:00

6.0 KiB

Raw Blame History

View Raw