环境搭建
首先在cmd或者anaconda promot中新建一个虚拟环境
conda create -n yolov7 python=3.9
- 注意:新建环境一定要把vpn关掉
激活环境,进入yolov7环境下,下载pytorch,进入pytorch官网进行下载,注意,要下载的cudn版本要小于本机的cudn版本,nvidia-smi版本为12.1,nvcc -v版本为12.0,我们选择要下载的cudn版本为11.6,同时我们新建的环境是python3.9,python和pytorch版本也有对应,如下图,我怕,我们选择pytorch1.12
activate yolov7
#在cmd命令下
conda activate yolov7
#promot下
安装完之后,我们可以在promot或者cmd进行验证