import torch print(torch.cuda.is_available()) >> True torch.cuda.get_device_name(0) >> 'GeForce GTX 1080 Ti' Pytorchに関する基礎知識を身に着けたいときには以下の書籍がおすすめだ. Tweet Pocket 関連