torch.device("cuda" if torch.cuda.is_available() else "mps" if torch.backends.mps.is_available() else "cpu")

Pytorch使うとき、こう書けば CUDA 使えるとき cuda、mps 使えるとき mps、それ以外は cpu ってデバイス指定できるかな？

device = torch.device("cuda" if torch.cuda.is_available() else "mps" if torch.backends.mps.is_available() else "cpu")