【PyTorch】GPUのメモリ不足でエラーになったときの対処方法。
PyTorch で深層学習していて、 GPUのメモリ不足でエラーが出てしまったので、対処方法のメモです。
エラーの内容は以下のような感じで「CUDA out of memory」となっています。
RuntimeError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0; 6.00 GiB total capacity;
4.55 GiB already allocated; 0 bytes free; 4.59 GiB reserved in total by PyTorch)
これは、GPU のメモリが解放できていないために起きているエラーのようです。
GPUメモリの解放は、コマンドプロンプトで行います。
コマンドプロンプトを立ち上げて、以下のコマンドで GPUメモリ情報を表示させます。
|
1 |
> nvidia-smi |
赤枠部分で、python.exe のプロセスが残っているために、メモリが解放されていないことが分かります。
以下のコマンドで、python.exe のプロセスをキルします。
|
1 |
> taskkill /pid プロセスID /f |
再度「nvidia-smi」コマンドでメモリが解放されていることが確認できます。
関連記事
-
-
【機械学習・手法比較】決定木とナイーブベイズを比較してみた。
同じデータを使って、教師有り機械学習手法の 決定木（Decision Tree）とナイーブベイズ（N
-
-
【機械学習】決定木（decision tree）について。
教師あり学習の一つである決定木（desicion tree）について勉強したことを書いていきます。
-
-
【Fashion-MNIST】ファッションアイテムのデータセットを使ってみた【TensorFlow】
今回は、機械学習用に公開されているデータセットの１つである「Fashion-MNIST」について紹介
-
-
【Chainer】手書き数字認識をしてみた【Deep Learning】
Chainerを用いて、ニューラルネットワークを構築し、手書き数字認識を行ったときのメモです。
-
-
機械学習の手法のまとめ。
機械学習は、「与えられた入出力事例をモデル化する行為」のことで、ディープラーニングなどで注目を集めて
-
-
【機械学習】 scikit-learn で不正解データを抽出する方法【Python】
Python の scikit-learn ライブラリを使って機械学習でテストデータを識別（２クラス
-
-
【Weka】フリーの機械学習ソフトをインストールする方法。
Weka は、ＧＵＩで使えるフリーの機械学習ソフトです。 https://ja.wikiped
-
-
【PyTorch】畳込みニューラルネットワークを構築する方法【CNN】
今回は、PyTorch を使って畳込みニューラルネットワーク（CNN）を構築する方法について紹介しま
-
-
【機械学習】 scikit-learn で精度・再現率・Ｆ値を算出する方法【Python】
今回は、２クラス分類で Python の scikit-learn を使った評価指標である、精度（P
-
-
【Weka】欠損データを自動的に補完するフィルタを使ってみた。
機械学習で用いるデータについてです。データは完璧なことに越したことはないが、通常は、ある属性の値が入
