栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

yoloV5环境配置过程记录

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

yoloV5环境配置过程记录

安装anaconda,从github下载yolov5的过程不再赘述

创建环境

conda create -n yolov5py38 python==3.8

进入环境

conda activate yolov5py38

进入yolov5文件夹

cd /d D:yoloyolov5-master //github克隆的yolov5-master所在路径

安装yoloV5依赖

pip install -r requirements.txt

或 conda install --yes --file requirements.txt

安装pytorch,检查GPU支持的Cuda版本

打开命令提示符,输入nvidia-smi

驱动版本511.23,CUDA版本11.6

进入pytorch官网选择配置

 复制安装命令

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

等待安装完毕,进入python,检查cuda是否可用。

import torch

print(torch.cuda.is_available())

检查GPU数量

print(torch.cuda.device_count())

返回当前GPU索引

print(torch.cuda.current_device())

检查GPU设备名称

print(torch.cuda.get_device_name(0))

打开pycharm,创建新工程,选择yolov5-master文件夹,环境选择刚才创建的yolov5py38

进入工程后,打开train.py,找到parse_opt函数,更改--device配置为0

parser.add_argument('--device', default='0', help='cuda device, i.e. 0 or 0,1,2,3 or cpu')

其余配置在整理数据集博文中说明

 

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/755001.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号