作りながら学ぶ PyTorchによる発展ディープラーニングの「2_objectdetection」の「2-7_SSD_training.ipynb」の実行エラーの解決方法

Question

前提・実現したいこと

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)

該当のソースコード

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

試したこと

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

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

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

Comments

[jbpb0]

> 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]

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

Answer 1

CUDA out of memory.

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