pytochプログラムが異常になったら占有の現存操作を削除します。

394 ワード

pytoch プログラム異常現存する

1-モデル変数の削除


del model_define

2-クリアCUDA cache


torch.cuda.empty_cache()

3-ステップ2（非同期）は一定時間が必要で、設定待ち時間を設定します。


time.sleep(5)

完全コードは以下の通りです。


del styler 
torch.cuda.empty_cache() 
time.sleep(5)

以上のpytouchプログラムは異常後、占有されている現存を削除する操作は小編集で皆さんに提供した内容を全部共有します。参考にしてほしいです。皆さんもよろしくお願いします。