Pytorch使うとき、こう書けば CUDA 使えるとき cuda、mps 使えるとき mps、それ以外は cpu ってデバイス指定できるかな？
Search pages and commands. Type `>` to search commands.