yolov7遇到的bug

遇到的一些bug

使用的原有配置YOLOv5的pytorvh环境,版本如下:

ubuntu=20.04
NVIDIA-SMI 460.106.00 Driver Version: 460.106.00 CUDA Version: 11.2
pytorch=1.7.0
1.test==

_pickle.UnpicklingError: STACK_GLOBAL requires str

解决: 参考办法
我直接把yolo5里面的data.yaml复制到yolo7里面,但是路径地下有两个缓存文件–train.cache,val.cache ,把这俩文件删掉就可以了
yolov7遇到的bug_第1张图片

2.train

RuntimeError: CUDA out of memory. Tried to allocate 2.96 GiB (GPU 0; 11.76 GiB total capacity; 1.31 GiB already allocated; 2.96 GiB free; 7.25 GiB reserved in total by PyTorch)

解决办法:参考方法
nvidia-smi
taskkill -PID 7392 -F(我电脑上没装taskkill)
kill 进程号
然后把训练命令的batch改为8,worker改为1

你可能感兴趣的:(bug,ubuntu,linux)