Python Pytorch Cuda Example Code