Solved by albanD in post #6 
              
              
                Now if you have args.use_cuda=True, your model will use the GPU. 
Keep in mind that if your model or input is very small, you might not use the gpu very efficiently and thus gpu usage will remain small. You need to perform large enough ops for the gpu to be properly used. Increasing the batch size i…