File size: 65 Bytes
3663dd0
 
 
1
2
3
import torch

print("CUDA available:", torch.cuda.is_available())