PyTorch が GPU を使用しているかどうかを確認するにはどうすればよいですか?nvidia-smi
コマンドは GPU アクティビティを検出できますが、Python スクリプト内から直接確認したいと思います。
ベストアンサー1
これらの関数は役立ちます:
>>> import torch
>>> torch.cuda.is_available()
True
>>> torch.cuda.device_count()
1
>>> torch.cuda.current_device()
0
>>> torch.cuda.device(0)
<torch.cuda.device at 0x7efce0b03be0>
>>> torch.cuda.get_device_name(0)
'GeForce GTX 950M'
これは次のことを物語っています。
- CUDA が利用可能で、1 つのデバイスで使用できます。
Device 0
は GPU を指しGeForce GTX 950M
、現在は PyTorch によって選択されています。