Verify CUDA in PyTorch

AI
python -c "import torch; print(torch.cuda.is_available(), torch.cuda.device_count())"