閉じる

PytorchがGPUを使っているかを確認する方法[Pytorch]

import torch

print(torch.cuda.is_available())
>> True

torch.cuda.get_device_name(0)
>> 'GeForce GTX 1080 Ti'

Pytorchに関する基礎知識を身に着けたいときには以下の書籍がおすすめだ.

Pocket

コメントを残す

あなたのメールアドレスは公開されません。必須項目には印がついています *

© 2020 CV・NLP TIPS | WordPress Theme: Annina Free by CrestaProject.