一:cpu版本
1. 无法安装opencv-python
改为opencv即可
2. 没有thop
官网下载安装包GitHub - Lyken17/pytorch-OpCounter: Count the MACs / FLOPs of your PyTorch model.,
解压后在sitepackage里安装
---路径输入cmd 同时按shift和鼠标右键即可打开当前路径下的终端
python setup.py install
参考:【YOLOv5问题记录】thop库的安装_啥都不会的材料菜鸡的博客-CSDN博客
3. ValueError: The requested array has an inhomogeneous shape after 1 dimensions
先debug到出问题那行,接着看shape,找到值none对应的图片(也就是出错的那张图),再到训练集文件夹中把对应图片的名字修改一下,重新运行即可解决问题。
二. cpu转gpu
1. torch torchvision下载相关(根据cuda版本 )
https://download.pytorch.org/whl/torch_stable.html
类似步骤2,进行安装,需要注意的是:
1). conda activate 相应的环境
2). cd到安装包目录
3). python -m pip install 包名(tab补全)
---否则pycharm 难以识别新的torch 和torchvision 包
2. TypeError: can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.
将报错代码self.numpy()改为self.cpu().numpy()即可
其他:
yolov5踩坑系列-1(error:‘Hardswish‘ object has no attribute ‘inplace‘)_ynxdb2002的博客-CSDN博客