df85439
1
2
3
4
5
6
7
8
import torch print(f"is availeble = {torch.cuda.is_available()}") print(f"device count = {torch.cuda.device_count()}") print(f"current device = {torch.cuda.current_device()}")
