Python Pytorch Cuda Example