初心者です。

fastai 1.0.42
torch 1.0.0
を使ってJupyter notebook上で画像分類をさせようと思っています。

学習前のnvidia-smi.exeを動かすとLostしておらず、torch.cuda.is_available()はTrueです。

learner.fit_one_cycle(n)でn回学習させた後、数秒待つとGPUがLostしてしまい、再起動しないとGPUが動いてくれません。

GPUがよくわかってないので、解決策を教えて欲しいです。

参考

学習前
画像の説明をここに入力
学習後
画像の説明をここに入力