File size: 46 Bytes
4008f8e
 
1
2
3
import torch
print(torch.cuda.is_available())