1. ホーム
  2. python

[解決済み] pytorchがGPUを使用しているかどうかを確認するにはどうすればよいですか?

2022-03-17 20:14:44

質問

を知りたいのですが。 pytorch が私のGPUを使用しています。を使って検出することが可能です。 nvidia-smi を書きたいのですが。 python スクリプトを使用します。

そのような方法はあるのでしょうか?

どのように解決するのですか?

これでうまくいくはずです。

import torch

torch.cuda.is_available()
>>> True

torch.cuda.current_device()
>>> 0

torch.cuda.device(0)
>>> <torch.cuda.device at 0x7efce0b03be0>

torch.cuda.device_count()
>>> 1

torch.cuda.get_device_name(0)
>>> 'GeForce GTX 950M'

これは、CUDAが利用可能であり、デバイス(GPU)の1つで使用できることを示しています。そして現在 Device 0 またはGPU GeForce GTX 950M が使用されています。 PyTorch .