File size: 90 Bytes
f35f8d1
 
 
1
2
3
4
import torch
print(torch.__version__)
print("CUDA available:", torch.cuda.is_available())