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

import torch

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

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

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

Pocket