Python 機械学習

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

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を無効化するには環境変数を設定すべし


KRSW

駆け出し機械学習エンジニア。機械学習、DB、WEBと浅く広い感じ。 Junior machine learning engineer. Not a specialist but a generalist who knows DB, WEB too.

役に立ったらシェアしてくれると励みになります。
If you felt this article is useful, please share.
にほんブログ村 IT技術ブログへ

-Python, 機械学習
-

Translate »
Social media & sharing icons powered by UltimatelySocial

Copyright© ITips , 2020 All Rights Reserved.