PytorchでGPU(CUDA)が使えるか否かを確認する方法

特にこれといったポイントもないんですけど備忘録として書きます。

pythonのインタラクティブシェルで以下をコピペするだけ。

import torch
print(torch.cuda.is_available())

Boolean型（TrueかFalseか）で帰ってくるのでわかりやすいですね。

CUDAインストールしてるのにFalseが出たときはバージョンを確認orCondaでインストールするで直ります。

以上！