Pytorch Cuda Example