Python 機械学習

TensorFlow、Kerasで機械学習する際にGPUを無効化する

TensorFlow、Kerasで機械学習する際にGPUを無効化する。
機械学習のコードがうまく動かない場合、コード中のバグをまず疑うが、中にはGPUとライブラリのバージョン不整合という場合もある。
そんなときにPython上で一旦GPUを無効化して動作をCPUに切り替える方法を記載する。


GPUの無効化

PythonでGPUを無効化する方法としては、環境変数でCUDAから利用できるデバイスを無効化すればよい。
具体的にはkerasのインポート前に以下の記述を追加する。

この操作を行うことでPythonからGPUを参照することができなくなり、CPUで計算が行われる。

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
import keras

Kerasをインポートする前に環境変数を設定すればいいみたいです。
Jupyter上でKerasを使うときにGPUを無効化する方法


まとめ

  • PythonでGPUを無効化するには環境変数を設定すべし


ITipsと同じようなブログを作る方法

ブログに興味がありますか?

もしブログに興味がある場合は↓このページ↓を参考にすれば、ITipsと同じ構成でブログを作ることができます

サーバー、ドメイン、ASPと【ブログに必要なものは全て】このページに書きました。
同じ構成でブログ作るのはいいけど、記事はマネしないでネ (TДT;)

ランキング参加中

にほんブログ村 IT技術ブログへ

他にもブログやSNSで紹介してくれると励みになります。

はてブのコメントで酷評されると泣きます(´;ω;`)

-Python, 機械学習
-

© 2022 ITips