実現したいこと
Pytorchの計算がGPU,CPUのどちらで計算されているかを確認したいです。
確認方法があればご教示ください。
前提
Pytorchで深層学習を試してみようと考えています。
以下のコードを入力しGPUが認識されていることは確認しました。
また、コード内でネットワークの重みやネットワークへの入力は.to(device)をつけてGPUで計算するようにしているつもりです。
(下記のコードからdeviceはcuda:0となっています)
しかし、タスクマネージャーではGPUが2%ほどしか使われていないと表示されます。
この時、なぜかCPUも8%程度であまり使われていません。
該当のソースコード
python
1print(torch.__version__) 2print(torch.cuda.is_available()) 3device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu") 4print(device) 5print(torch.cuda.get_device_name(0))
上記の出力
1.12.0+cu116
True
cuda:0
NVIDIA GeForce RTX 3080
試したこと
タスクマネージャーにcudaが表示されていなかったため、
以下手順で表示されるようにしましたが、
やはり使用率は2%程度です。
https://stealthinu.hatenadiary.jp/entry/2022/07/30/183632?utm_source=feed
補足情報(FW/ツールのバージョンなど)
cuda version 11.6
pytorch 1.12.0+cu116
OS windows11
回答1件
あなたの回答
tips
プレビュー