Python Pytorch Cuda Compatibility Test