Mac での GPU を使用した Pytorch トレーニング

Mac で GPU を駆動するコードを次のように変更する必要があります。

device = torch.device("mps")

おすすめ

転載: blog.csdn.net/qq_45191169/article/details/128211090