File size: 150 Bytes
4f7ba4c
 
645d766
 
 
 
1
2
3
4
5
6
7
import torch

if torch.cuda.is_available():
    print("✅ GPU is available:", torch.cuda.get_device_name(0))
else:
    print("❌ No GPU detected!")