用jupyter跑yolov5的代码，报错out of memory

train.py参数设置

报错RuntimeError: CUDA error: out of memory

因为自己的电脑没有GPU,老师给了一个jupyter账号。代码在自己的电脑上可以运行，但在jupyter上会报这种错，batch_size修改到1还是会报错，实在是不知道哪里出了问题

可以参考一下https://stackoverflow.com/questions/54374935/how-to-fix-this-strange-error-runtimeerror-cuda-error-out-of-memory

可能别人也在用这个gpu
确定一下是否真的是内存问题

cuda内存都是0？

估计是爆显存了, 看一下后台进程, 把GPU占用率高的进程能关就关

应该是woek_mode改为=0吧

预训练模型太大了选个小点的前两行和图片那行注意配套哦