SecurityDemo / app.py
mkhodary101's picture
Update app.py
645d766 verified
raw
history blame
150 Bytes
import torch
if torch.cuda.is_available():
print("✅ GPU is available:", torch.cuda.get_device_name(0))
else:
print("❌ No GPU detected!")