在windows调了一周的yolo环境
在ubuntu调试了一个小时
…
只因—不知道torch和torchvision怎么装(当时在咸鱼找大佬帮我装的时候,我大为震惊…)
(我不是在pycharm运行pip install -r requirement.txt
了,为什么还是不行?)
大佬用了一个办法—在这个网站上找了pytorch和torchvision然后install下来
download.pytorch.org/whl/torch_stable.html
然后我乖乖交了补课费
pytorch和torchvision要官方单独安装
去官方网站 PyTorch
在表格里面选择自己需要的
然后进行pip下载
如果网不好就
pip install XXX -i https://pypi.mirrors.ustc.edu.cn/simple/
//中科大的下载速度yyds
教程的连接: pip 使用国内镜像源 | 菜鸟教程 (runoob.com)
其他bug目录
运行train.py,git error的报错
因为被屏蔽了,这时候最好把pt模型文件下载下来,然后放在main主目录下
原理:git会从官网下载模型,然后根据模型进行训练
那么我们直接下载模型到主目录,就可以省略这个步骤
例如
‘https://github.com/WongKinYiu/yolov7/releases/tag/v0.1’
yolov7 模型权重下载网址
运行时候需要加上参数
这个时候强推 pycharm 里面对文件编辑的设置,可以避免我重新在命令行输入参数,
我继续说我后来装cuda 和cudnn的事情
- pycharm/vscode 都不太能够运行 train+gpu加速,容易报错
- 需要