🎄teratailクリスマスプレゼントキャンペーン2024🎄』開催中!

\teratail特別グッズやAmazonギフトカード最大2,000円分が当たる!/

詳細はこちら
強化学習

強化学習とは、ある環境下のエージェントが現状を推測し行動を決定することで報酬を獲得するという見解から、その報酬を最大限に得る方策を学ぶ機械学習のことを指します。問題解決時に得る報酬が選択結果によって変化することで、より良い行動を選択しようと学習する点が特徴です。

Anaconda

Anacondaは、Python本体とPythonで利用されるライブラリを一括でインストールできるパッケージです。環境構築が容易になるため、Python開発者間ではよく利用されており、商用目的としても利用できます。

機械学習

機械学習は、データからパターンを自動的に発見し、そこから知能的な判断を下すためのコンピューターアルゴリズムを指します。人工知能における課題のひとつです。

Unity

Unityは、Unity Technologiesが開発・販売している、IDEを内蔵するゲームエンジンです。主にC#を用いたプログラミングでコンテンツの開発が可能です。

Q&A

解決済

1回答

4269閲覧

Unity ML-agentsでの学習でGPUが使用されていない

tspost

総合スコア4

強化学習

強化学習とは、ある環境下のエージェントが現状を推測し行動を決定することで報酬を獲得するという見解から、その報酬を最大限に得る方策を学ぶ機械学習のことを指します。問題解決時に得る報酬が選択結果によって変化することで、より良い行動を選択しようと学習する点が特徴です。

Anaconda

Anacondaは、Python本体とPythonで利用されるライブラリを一括でインストールできるパッケージです。環境構築が容易になるため、Python開発者間ではよく利用されており、商用目的としても利用できます。

機械学習

機械学習は、データからパターンを自動的に発見し、そこから知能的な判断を下すためのコンピューターアルゴリズムを指します。人工知能における課題のひとつです。

Unity

Unityは、Unity Technologiesが開発・販売している、IDEを内蔵するゲームエンジンです。主にC#を用いたプログラミングでコンテンツの開発が可能です。

0グッド

0クリップ

投稿2020/12/09 11:37

編集2020/12/10 12:14

前提・実現したいこと

ML-agentsでの強化学習でtensorflow-GPUを正しく機能させたい

発生している問題

現在ML-agentsとAnacondaを用いて強化学習を勉強しており,GPUによる処理で学習を高速化できるとネットで見たためcuda,cudnnとtensorflow-gpuを入れてみたのですが,タスクマネージャーを見てもGPUの使用率は全く上がらずtensorflow-gpuを導入する前と学習速度やCPU負荷が全く変わらない状況です.
学習開始前に一枚目のような表示が出ており,unityのスタートボタンを押すとCPU処理の時とは違い5分ほどunityが処理待ち?で停止(フリーズ)してから学習を開始し,またGPUの名前も表示されるのでGPUは認識されていると思うのですがタスクマネージャーを見てもCPUの使用率が上がるばかりでGPUのメモリ使用率は全く上がっておらず学習速度も向上しているとは思えない状態です.
どのようにすればGPUでの処理が可能になるのでしょうか?

以下にanacondaの表示画面を添付します.
1枚目(yaml実行直後):
イメージ説明

2枚目(unityのスタートボタンを押して5分後):
イメージ説明

3枚目(さらにそれから1分半後,この直後に学習開始):
イメージ説明

試したこと

CUDA,CUDnnのバージョン確認、環境変数の見直し、解説サイトの閲覧

補足情報(FW/ツールのバージョンなど)

CUDAのバージョン:10.1
CUDnnのバージョン:CUDAのv10.1に対応したv7.6.5
tensorflow-gpu:2.3.1
CPU:AMD Ryzen 7 1700
GPU:RTX 3060ti
CUDA,CUDnnのインストール先:Cドライブ(起動ドライブ)
Unity,anacondaのインストール先:Hドライブ

###試したことと追加の質問(2020/12/10追記)
上記の環境で

from tensorflow.python.client import device_lib device_lib.list_local_devices()

を実行したところデバイスタイプがCPUとなっていたためtensorflowのバージョンを落としてみたところGPUの認識に成功しました.
またunity上で100以上のオブジェクトを並べて並列学習を行っていたためもしやCPUの負荷はこれが原因ではないかとサンプルプログラムのほうで改めて学習をさせてみたところCPUの負荷も軽くGPUも使用していたためCPUの負荷はUnityの大量描画が原因であることがわかり,おそらくGPUでの学習に成功しました

しかし

from tensorflow.python.client import device_lib device_lib.list_local_devices()

や学習プログラムを実行した際,GPUを認識するのに10分ほどかかるのですがこれを早めることはできないのでしょうか?また学習中のタスクマネージャーがサンプルと大量描画それぞれのpyhon使用ピーク時で以下のようになっていたのですがGPUによる学習の負荷はこんなものなのでしょうか?

サンプルプログラム:
イメージ説明
大量描画:
イメージ説明

気になる質問をクリップする

クリップした質問は、後からいつでもMYページで確認できます。

またクリップした質問に回答があった際、通知やメールを受け取ることができます。

バッドをするには、ログインかつ

こちらの条件を満たす必要があります。

fiveHundred

2020/12/10 00:16

(ML-agentsは詳しくないので、検討違いだったらすいません) GPUでないtensorflowはインストールされていますか? もしそうであれば、そちらの方が動いている可能性もあるので、アンインストールしてみるか、それが無い別の仮想環境を用意してみるとどうでしょうか?
guest

回答1

0

自己解決

CUDA,cudnn,tensorflowのバージョンを再度見直したところGPUの認識に成功しました
学習速度の向上はあまり体感できませんでしたがunityを見たところunityとml-agentsのスクリプトの処理の問題のようなのでそちらを見直したいと思います

投稿2020/12/24 04:40

tspost

総合スコア4

バッドをするには、ログインかつ

こちらの条件を満たす必要があります。

あなたの回答

tips

太字

斜体

打ち消し線

見出し

引用テキストの挿入

コードの挿入

リンクの挿入

リストの挿入

番号リストの挿入

表の挿入

水平線の挿入

プレビュー

15分調べてもわからないことは
teratailで質問しよう!

ただいまの回答率
85.36%

質問をまとめることで
思考を整理して素早く解決

テンプレート機能で
簡単に質問をまとめる

質問する

関連した質問