AIで2次元キャラを生成するサービス開始!

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

この記事は約1分で読めます。

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

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

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

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

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

以上!

コメント

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