How to check if pytorch is using the GPU? How to check if pytorch is using the GPU? python python