【特集一覧】最新ガジェットから独占取材まで!
オタクに捧ぐ公式Twitter、始動(告知バナー)

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

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

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

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

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

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

以上!

コメント

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