初心者です。
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がよくわかってないので、解決策を教えて欲しいです。
学習前
学習後