yolov过程配置问题解决
- 计算机科学
- 2024-03-08
- 435热度
- 0评论
最近又在捣鼓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