作りながら学ぶ PyTorchによる発展ディープラーニングの「2_objectdetection」の「2-7_SSD

前提・実現したいこと

PyTorchによるSSDを用いた物体検出の訓練について

「作りながら学ぶ　PyTorchによる発展ディープラーニング」
の第2章　「2-7_SSD_training.ipynb」
において「学習・検証を実施する」以下のセルを実行した際に以下のエラーメッセージが発生しました。

発生している問題・エラーメッセージ

RuntimeError: CUDA out of memory. Tried to allocate 2.00 MiB (GPU 0; 8.00 GiB total capacity; 2.47 GiB already allocated; 0 bytes free; 2.79 GiB reserved in total by PyTorch)

該当のソースコード

PyTorch
1 学習・検証を実行する
2num_epochs= 10  
3train_model(net, dataloaders_dict, criterion, optimizer, num_epochs=num_epochs)

試したこと

ネット検索をしてみたがい解決方法がわからなかった。

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

本のGIT　URL：https://github.com/YutaroOgawa/pytorch_advanced

jbpb0

2021/04/12 03:59 編集

> CUDA out of memory. GPUのメモリーが不足してます GPUを使ったPythonのプロセスが残ってませんでしょうか？ https://betashort-lab.com/%E3%83%97%E3%83%AD%E3%82%B0%E3%83%A9%E3%83%9F%E3%83%B3%E3%82%B0/python/jupyternotebook%E3%81%A7gpu%E3%81%AE%E3%83%A1%E3%83%A2%E3%83%AA%E3%82%92%E8%A7%A3%E6%94%BE%E3%81%99%E3%82%8B%E6%96%B9%E6%B3%95-windows%E7%B7%A8/

Saba3

2021/04/12 05:12

確認したところ、ご指摘通りJupter内の別のipynbがRunning状態でした。全てShutdownにして再度実行してみるとエラーはでなくなりました。教えていただいてありがとうございました！

行動規範の内容に同意します

回答1件

ベストアンサー

CUDA out of memory.

GPUのメモリーが不足してます
GPUを使ったPythonのプロセスが残ってたら、止めてみてください
JupyterNotebookでGPUのメモリを解放する方法-Windows編

投稿2021/04/12 06:44

jbpb0

総合スコア7651

あなたの回答

tips

プレビュー

行動規範の内容に同意します

質問の解決につながる回答をしましょう。サンプルコードなど、より具体的な説明があると質問者の理解の助けになります。また、読む側のことを考えた、分かりやすい文章を心がけましょう。

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

ただいまの回答率
85.48%

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

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

質問する

質問をすることでしか得られない、回答やアドバイスがある。

15分調べてもわからないことは、質問しよう！