Ubuntuでの機械学習準備について（GPU）

Question

Ubuntu18.04lts
python3.6.7
GeForce RTX 2060

機械学習用の環境を構築するため、GPUの設定をしたのですが作動していないように見えます。

ターミナルに"nvidia-smi"と入力すると、
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 440.44 Driver Version: 440.44 CUDA Version: 10.2 |
|-------------------------------+----------------------+----------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
|===============================+======================+======================|
| 0 GeForce RTX 2060 Off | 00000000:01:00.0 Off | N/A |
| 0% 57C P0 17W / 160W | 0MiB / 5931MiB | 0% Default |
+-------------------------------+----------------------+----------------------+

+-----------------------------------------------------------------------------+
| Processes: GPU Memory |
| GPU PID Type Process name Usage |
|=============================================================================|
| No running processes found |
+-----------------------------------------------------------------------------+
と表示されます。

どのようにすればGPUが作動するのでしょうか。

試したこと
１．nvidia関係のファイルを全て削除してから再インストール
２．sudo vi /etc/modprobe.d/blacklist-nou.confの編集

以上です。よろしくお願いいたします。

追加・修正
watch -n1 "nvidia-smi"コマンドを確認しながらnumpyで作業しました。
これはGPUが使われているのか確認方法がわからなかったためです。

Comments

[tiitoi]

どのように動作していないと判断されたのでしょうか？
GPU はなにも処理していなければ、0%になっていても別におかしくないです。

Answer 1

watch -n1 "nvidia-smi"コマンドを確認しながらnumpyで作業しました。

numpy の演算はすべて CPU で行われますので、GPU は使用されません。
GPU を使用するライブラリ、ソフトウェアを使わない限り、GPU の恩恵は受けられません。
数値計算であれば、cupy、Pytorch、TensorFlow など GPU を使うように設計された数値計算ライブラリを使う必要があります。

GPU を使用することで性能向上が見込める例

• ゲーム
• 動画のエンコード
• ディープラーニング

Comments

[mako10]

回答頂きありがとうございます。
tensorflowとkerasを使おうと考えていたのですが、導入前に動作確認をしようと考えていました。
cupyで確認してみます。
ありがとうございました。