YOLOV+pytorch+win10+CPU环境配置

Step 1:下载github YOLOV3源码

链接:https://github.com/ultralytics/yolov3

Step 2:配置CPU+pytorch版本环境

Win+R启动cmd,在命令提示符内输入以下命令,创建一个新环境:

conda create –n yolov-pytorch python=3.7

激活环境:

activate yolov-pytorch

Step 3:安装YOLOV3仓库中requirements.txt的安装包

注意如果是CPU版本,那么先不装torch和torchvision
YOLOV+pytorch+win10+CPU环境配置_第1张图片

pip install -r requirements.txt

安装CPU版本的torch与torchvision,这里参考该博客

# CPU only
pip install torch==1.7.0+cpu torchvision==0.8.1+cpu -f https://download.pytorch.org/whl/torch_stable.html

注意:配置环境的时候如果用的国内镜像,记得把VPN关了

成功配置如下:
YOLOV+pytorch+win10+CPU环境配置_第2张图片YOLOV+pytorch+win10+CPU环境配置_第3张图片

Step 4:训练自己数据集教程

百度、谷歌出来的YOLO教程大多是老版的1,跟不上github仓库的更新频率,导致很多代码都用不了,推荐一个教程让大家少走弯路!
yolov5训练自己数据集,小白也能学会,详细教学

你可能感兴趣的:(计算机视觉,pytorch,python,深度学习)