Python: Keras/TensorFlow で GPU のメモリを必要な分だけ確保する - CUBE SUGAR CONTAINER

テクノロジーカテゴリーの変更を依頼記事元:

blog.amedama.jp

32 usersがブックマークコメント

コメント

1

記事へのコメント1件

注目コメント
新着コメント

ST_ha1cyon “Keras のバックエンドに TensorFlow を使う場合、デフォルトでは一つのプロセスが GPU のメモリを全て使ってしまう。” これが一番の原因だった・・・openposeのpythonラッパーはこれが書かれてるのね

2019/01/17 リンク

注目コメント算出アルゴリズムの一部にLINEヤフー株式会社の「建設的コメント順位付けモデルAPI」を使用しています

規約違反を報告

アプリのスクリーンショット

いまの話題をアプリでチェック！

バナー広告なし
ミュート機能あり
ダークモード搭載

アプリをダウンロード

関連記事

{{ total_bookmarks_with_user_postfix }}{{ root_title }}

Python: Keras/TensorFlow で GPU のメモリを必要な分だけ確保する - CUBE SUGAR CONTAINER

Keras のバックエンドに TensorFlow を使う場合、デフォルトでは一つのプロセスが GPU のメモリを全て使... Keras のバックエンドに TensorFlow を使う場合、デフォルトでは一つのプロセスが GPU のメモリを全て使ってしまう。今回は、その挙動を変更して使う分だけ確保させるように改めるやり方を書く。環境には次のようにしてセットアップした Ubuntu 16.04 LTS を使っている。 blog.amedama.jp サンプルとして動作させるアプリケーションには Keras が提供している MNIST データセットを CNN で認識するものを使う。まずはこれをダウンロードしておこう。同時に、セッションをクリアするパッチも追加しておく。 $ wget https://raw.githubusercontent.com/fchollet/keras/master/examples/mnist_cnn.py $ echo 'K.clear_session()' >> mnist_c

ブックマークしたユーザー

satojkovic2019/10/01
k-kty2019/02/25
ST_ha1cyon2019/01/17
dann2018/12/04
Gln2018/07/29
koma_g2018/02/04
yukimori_7262017/10/02
kuedajp20102017/09/08
pidekazu2017/08/22
kjw_junichi2017/06/14
labunix2017/06/09
hohoho_ho20052017/06/09
ryshinoz2017/06/09
masayoshinym2017/06/09
takayuki0072017/06/09
manabou2017/06/08
s_nkmc2017/06/08
Jyuichi2017/06/08

同じサイトの新着

同じサイトの新着をもっと読む

いま人気の記事

いま人気の記事をもっと読む

いま人気の記事 - テクノロジー

いま人気の記事 - テクノロジーをもっと読む

新着記事 - テクノロジー

新着記事 - テクノロジーをもっと読む

設定を変更しましたx