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
版权声明:
作者:lichengxin
链接:https://www.techfm.club/p/13970.html
来源:TechFM
文章版权归作者所有,未经允许请勿转载。
THE END
二维码
共有 0 条评论