1、下载并解压:Yolov7 Gitee源代码
2、下载模型,一般用YOLOv7即可,放到Yolov7根目录下。
3、注释掉requirements.txt中的两行。
4、创建anaconda环境,Python版本3.6~3,.8都行,然后输入命令安装依赖。
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
5、安装成功之后先卸载torch,因为这是cpu版本的。如果你没有安装过cuda和cudnn,那只能使用cpu版本。
6、到ptorch镜像网站上下载wheel安装包,选择适合自己cuda版本以及python版本的,torch和torchvision版本要对应。
https://download.pytorch.org/whl/torch_stable.html
7、先安装torch。
pip install torch-1.2.0-cp37-cp37m-win_amd64.whl
8、再安装torchvision。
pip install torchvision-0.4.0-cp37-cp37m-win_amd64.whl
9、进入python,import无报错,且可以调用。
10、在Pycharm中设置工程的编译器。
11、然后在终端输入命令python detect.py即可调用GPU推理。