PyTorch Forums
[SOLVED] Make Sure That Pytorch Using GPU To Compute
vision
pinocchio
(Rene Sandoval)
February 10, 2020, 9:05pm
29
is there some sort of internal flag I can check to see if things are properly placed in GPU?
What is the difference between doing `net.cuda()` vs `net.to(device)`?
show post in topic