>>106407041
DisTorch2 Model Final Device/Layer Assignments
--------------------------------------------------
Device Layers Memory (MB) % Total
--------------------------------------------------
cuda:0 (<0.01%) 283 2.74 0.0%
cuda:0 224 7486.60 54.9%
cpu 181 6138.56 45.0%

Can't even get a single step done like this because the CPU is working its ass off while my GPU basically just idles. Doesn't seem like a good thing for single GPU setups.