PytorchでGPU(CUDA)が使えるか否かを確認する方法TwitterFacebookLINEコピーコメント 2021.02.25特にこれといったポイントもないんですけど備忘録として書きます。pythonのインタラクティブシェルで以下をコピペするだけ。import torchprint(torch.cuda.is_available())Boolean型(TrueかFalseか)で帰ってくるのでわかりやすいですね。CUDAインストールしてるのにFalseが出たときはバージョンを確認orCondaでインストールするで直ります。以上!
コメント