PyTorchでGPUが使えるか確認する

※自分用メモ

PyTorchでGPUが使えるか確認する方法をメモしておきます。

結論としては以下のコードを実行すれば確認できます。

import torch

is_cuda = torch.cuda.is_available()

if is_cuda:
    print("gpu")
else:
    print("cpu")

torch.cuda.is_available() でcudaを認識しているかわかります。

コメント

タイトルとURLをコピーしました