Starting with pytoch - check installation

Sorry for the noby question. I installed torch 1.1 on my windows machine.
I am not sure if it is ok since “torce.cuda.is_available()” return true but all examples I try seems to not run on the gpu.

Is there a simple code (A simple torch loop for example) I can run just to see that the gpu is loaded so I can verify that torch can really run on gpu?

Thanks in advanced.


1 Like
1 Like