import torch print(torch.__version__) device = ( "cuda" if torch.cuda.is_available() else "mps" if torch.backends.mps.is_available() else "cpu" )
import torch torch.cuda.is_available() torch.cuda.device_count() torch.cuda.get_device_name(0)
Seonglae Cho
Seonglae Choimport torch print(torch.__version__) device = ( "cuda" if torch.cuda.is_available() else "mps" if torch.backends.mps.is_available() else "cpu" )
import torch torch.cuda.is_available() torch.cuda.device_count() torch.cuda.get_device_name(0)