YOLOv5训练coco128数据集流程

一、安装pytorch
1、创建虚拟环境
         使用Anaconda安装,建议先创建一个虚拟环境。启动Anaconda Prompt,在命令行输入:
         conda create -name torch python==3.7.11
2、安装pytorch
         进入pytorch官网,根据需要选择系统、安装包、语言和计算平台。
         通过conda安装CPU版pytorch:
         conda install pytorch torchvision torchaudio cpuonly -c pytorch
         安装其他版本pytorch:点击Previous versions of PyTorch,找到更早的版本,比如1.8.0。在Anaconda命令行输入:
         conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cpuonly -c pytorch

YOLOv5训练coco128数据集流程最先出现在Python成神之路

版权声明:
作者:lichengxin
链接:https://www.techfm.club/p/13970.html
来源:TechFM
文章版权归作者所有,未经允许请勿转载。

THE END
分享
二维码
< <上一篇
下一篇>>