Python check PyTorch GPU

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