Pytorch Check Cuda Availability