YOLOv5环境配置

yolov5环境配置

因为项目要求,本文主要使用anaconda ,pycharm,安装包使用GPU版本
暂时更新到安装包,所以其实一点yolov5的内容都还莫得

安装anaconda和pytorch

  1. anaconda可以从镜像源网站下载
    阿里云
    http://mirrors.aliyun.com/pypi/simple/
    中国科技大学
    https://pypi.mirrors.ustc.edu.cn/simple/
    豆瓣(douban)
    http://pypi.douban.com/simple/
    清华大学
    https://pypi.tuna.tsinghua.edu.cn/simple/
    中国科学技术大学
    http://pypi.mirrors.ustc.edu.cn/simple/
    (记住这几个网站,一会要考)
    记得检查环境变量
  2. pycharm从官网下载即可,但只有社区版,专业版需学生认证后可免费下载
    (一年一认证,一次管一年)
    官网
    学生认证申请
  3. 创建新环境
    安装完之后可以顺便把anaconda换一下镜像源(记得在基本环境,就是base环境安装)
    就是打开命令行(win+R,输入cmd。或者在开始菜单找到anaconda的prompt,然后输入下列命令)
    (命令行输入代码时尽量不要直接复制粘贴,一行一行输入)
conda config --remove-key channels
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.bfsu.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes
pip config set global.index-url https://mirrors.ustc.edu.cn/pypi/web/simple

然后创建新环境:(我这里选的mask_yolov5代表环境名,创建时,可以参考自己的项目名或创建环境的目的)

conda create -n mask_yolo5 python==3.8

激活新环境

conda activate mask_yolo5

然后用pycharm打开项目文件夹,在pycharm的右下角可以找到添加解释器:
pycharm界面
选择刚才的环境添加

安装相关包

如果项目中含有requirements.txt,说明,,,这个人是个好人(bushi
说明可以直接安装大部分包,pycharm也会有提示“是否安装requirements里的包”。但仍有几个包需要手动安装(哪里报错按哪个)
方式:

cd 项目文档路径
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple --user

笔者主要在tensorflow和pytorch安装时,屡屡碰壁,所以主要说这两个包的安装

1. tensorflow的gpu版本安装

首先,大胆尝试直接安装(记得激活环境)

pip install tensorflow-gpu==2.8.0

然后我报错了,,,第一次是TypeError: Descriptors cannot not be created directly. If this call came from a _pb2。按照网上教程把protobuf降级

pip uninstall protobuf
pip install protobuf==3.19.0

然后报错AttributeError: module ‘keras.utils.generic_utils‘ has no attribute ‘populate_dict_with_module。是keras与tensorflow版本不匹配(我真的会谢
下一个匹配的版本吧,,,
版本对应网站
然后检查一下: 先进入python(就是激活环境,然后输入python)

import tensorflow as tf
print(tf.test.is_gpu_available())

输出是true就可以

2. 安装pytorch

pytorch需要注意的是,用conda直接从镜像源下载,下的是cpu版本(诡计多端的镜像源 )所以我们需要从官网下载whl文件,然后安装
这里要感谢这位博主!!下载whl文件教程
然后最后再测试一下就可以了

python
import torch
torch.cuda.is_available()
配置 YOLOv5 的运行或开发环境可以根据不同的操作系统和硬件平台采用多种方式来完成。以下是几种常见的方法: ### 基于 Windows 配置 YOLOv5 开发环境 1. **创建 Python 虚拟环境** 使用 Anaconda 创建独立的虚拟环境是一个推荐的做法,这样可以避免不同项目之间的依赖冲突。打开 Anaconda Prompt 并执行以下命令: ```bash conda create -n yolov5 python=3.8 conda activate yolov5 ``` 2. **安装 PyTorch 和其他必要的库** 根据您的 GPU 支持情况选择合适的 CUDA 版本进行安装。对于支持 NVIDIA 显卡的用户来说,可以选择带有 `cudatoolkit` 的版本: ```bash conda install pytorch torchvision torchaudio cudatoolkit=11.0 ``` 如果网络速度较慢,则可以通过切换到国内镜像加速下载过程。 3. **克隆官方仓库并准备数据集** 接下来需要从 GitHub 上获取最新的 YOLOv5 代码,并准备好用于训练或者测试的数据集文件夹结构。 4. **验证安装是否成功** 可以通过运行一些预定义脚本来确认一切设置无误,比如尝试加载模型并对图像进行推理操作。 ### 利用 Docker 构建跨平台解决方案 如果您希望快速部署而不必担心底层系统的差异性问题,那么使用 Docker 是一种非常有效的方式[^2]。只需一条简单的命令即可启动一个含所有必需组件的工作空间: ```bash docker run -it --name yolov5 --gpus all -v /path/to/local/directory:/yolov5 ultralytics/yolov5:latest bash ``` 这条指令不仅会拉取最新版的基础映像,还将挂载本地目录以便后续处理自己的数据集或其他资源。 ### 对 Jetson Nano 用户的一些建议 针对特定嵌入式设备如 NVIDIA Jetson Nano,在构建适用于该类低功耗高性能计算模块的应用程序时需要注意其独特的软硬兼容需求。虽然理论上讲大多数标准流程都适用,但在实际应用过程中可能还需要额外调整某些参数以达到最佳性能表现[^1]。 ---
评论 1
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值