yolov过程配置问题解决

最近又在捣鼓yolov的相关工作了,配置环境中遇到一些小问题记录一下

1 AttributeError: module ‘numpy‘ has no attribute ‘int‘.

问题原因:新版本numpy种已经没有np.int的描述了

解决方法:

a 降低numpy的版本

pip uninstall numpy
pip install numpy

b 使用第二种表述

np.int改为np.int_

还是用第二种方法比较得劲

2. build target indices.append((b, a, gj.clamp(0, gain[3] - 1), gi.clamp(0, gain[2] - 1)

讲yolov的loss文件

gain = torch.ones(7, device = device)

修改为:

gain = torch.ones(7, device = device).long()

3. 一个酷炫的内存监视系统

pip install nvitip
nvitop

image-20240308210254525