When I specify to run my code on specific GPU(GPU-2), it uses that GPU specified but also creates subprocesses of around 500MB in GPU-0. Assigned GPU using: network.cuda(2) image.cuda(2) Moved data to GPU using: mydata.to(2) Pytorch version used: 0.4 Assigned GPU: Green (GPU-2) Subprocesses: …

GPU utilization in Pytorch 0.4

InnovArul (Arul) September 12, 2018, 10:19am 2

This issue is fixed in version 0.5.0

1 Like

show post in topic

Powered by Discourse, best viewed with JavaScript enabled