yolov7环境配置(pytorch-gpu配置)

1 anaconda 安装(去官网下载即可)

要会使用anaconda的环境,下面的实验基本都是在anaconda中管理python环境

2 虚拟环境下配置代码环境

这样做的目的主要是不把conda的base环境弄得混乱不堪,弄错或者有问题在虚拟环境中直接删除重来就行,conda的虚拟环境位于anaconda目录下的envs目录里
-n 后面是指定虚拟环境名字,可以随意起

conda create -n yolo python==3.7

激活环境后,这里可以看到prompt终端前面的括号内容变化,从(base)-> (yolo)

conda activate yolo

3 pytorch 安装

查看自己电脑支持的cuda最高版本(在cmd终端中查看):

nvidia-smi

cuda版本

下面两种方法并没有说安装cuda和cudann,理论需求的cuda版本是10.0,实际情况下,安装pytorch会包含cuda等相关驱动(一些博客说的的,实践证明确实不需要再安装什么cuda等,也省去了版本难以对齐的问题),或者说现在大部分Win10的电脑已经有相关驱动,总之,下载之后是可以直接用的
方法一(一般情况下该方法速度比较快):
去这个网站下载轮子:http://download.pytorch.org/whl/torch/
需要版本:torch=1.2.0 torchvision=0.4.0
下载后,在终端用cd命令到whl下载的目录,之后pip install + 轮子文件名
方法二(官网下载,速度可能快可能慢)
文件主体大小约800Mb

pip install torch=1.2.0 torchvision=0.4.0 -f https://download.pytorch.org/whl/torch_stable.html

验证安装成功(在yolo环境下,及刚刚上面创建的虚拟环境下,运行一段Python程序,看看是否报错):

import torch
print(torch.cuda.is_available())

输出结果为TRUE说明安装成功

需求文档

在yolo环境下conda 或者pip安装对应版本即可,下载速度慢可以加上参数 -i 指定源

  • 阿里云 https://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/
scipy==1.2.1
numpy~=1.21.6
matplotlib==3.1.2
opencv_python==4.1.2.30
torch==1.2.0
torchvision==0.4.0
tqdm~=4.64.1
Pillow==8.2.0
h5py==2.10.0
##格式参考如下
pip install scipy==1.14.0  -i https://pypi.tuna.tsinghua.edu.cn/simple/

也可以去这个网站下载对应包安装:https://pypi.org/

### 回答1: b'yolov4环境搭建pytorch(gpu)'指的是在使用pytorch深度学习框架并使用GPU计算的情况下,配置yolov4目标检测模型的开发环境。这可以通过在计算机上安装pytorch和CUDA加速库来实现。同时,还需要下载yolov4的代码和权重文件,并对其进行配置和调试,确保能够成功使用GPU进行模型训练和测试。 ### 回答2: 步骤一:安装依赖库 在进行yolov4环境搭建之前,我们需要先安装以下依赖库: * PyTorch:安装命令为:pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu111/torch_stable.html * NumPy:安装命令为:pip install numpy * OpenCV:安装命令为:pip install opencv-python * Cython:安装命令为:pip install cython * Matplotlib:安装命令为:pip install matplotlib 步骤二:下载YOLOv4代码 在安装完以上依赖库后,我们需要下载YOLOv4的代码。可以从GitHub上进行下载,将其解压缩到本地的项目目录中即可使用。 步骤三:编译和安装YOLOv4 在安装好依赖库和下载好YOLOv4代码后,我们需要执行以下步骤进行编译和安装。 1.进入yolo文件夹,执行以下命令: * python setup.py build_ext(执行此命令会创建一个 build文件夹) * python setup.py install(执行此命令之后,会将yolo安装到系统中) 2. 下载yolov4.weights文件,将其保存在YOLOv4目录下的weights文件夹中。 步骤四:测试YOLOv4 在安装完毕之后,我们需要进行一些测试,以确保YOLOv4的可用性。 1. 进入YOLOv4目录,执行以下命令: * python detection.py --image-path data/samples/zidane.jpg --cfg cfg/yolov4.cfg --weights weights/yolov4.weights 2. 如果运行成功,将会看到zidane.jpg的检测结果。 以上是关于yolov4环境搭建的步骤解析。需要注意的是,搭建过程中可能会遭遇一些错误和问题,需要根据实际情况进行调整。此外,安装过程需要一定的技术基础,对于没有经验的初学者来说,可能需要较长时间的摸索和学习。 ### 回答3: yolov4是一种高效的目标检测算法,对于需要进行大规模物体识别的场景非常有用。为了更好地应用yolov4算法,我们需要在本地搭建pytorch(gpu)实验环境以进行训练与验证。 以下是yolov4环境搭建pytorch(gpu)的步骤: 步骤1:安装Anaconda 在Windows系统下,我们可以前往anaconda官网下载对应安装包并进行安装,安装完成后会自动配置好conda的环境。 步骤2:创建Python虚拟环境 通过conda创建一个Python虚拟环境,并安装pytorch及其他必要的依赖库: conda create -n yolov4 python=3.7 conda activate yolov4 conda install pytorch torchvision cudatoolkit=10.2 -c pytorch pip install opencv-python 步骤3:下载YOLOv4-tiny模型权重 可以在Github上下载yolov4-tiny的模型权重,将下载好的权重文件放在项目代码的根目录下。 步骤4:编写yolov4算法实现代码 通过pytorch编写代码实现yolov4算法,并利用GPU进行加速。 步骤5:使用自己的数据集进行模型训练 将自己的数据集按预设格式进行制作,然后通过上面搭建好的环境进行模型训练,并保存模型。 步骤6:使用模型进行目标检测 使用已经训练好的模型进行目标检测,根据需要进行调整和优化。最后,可以将模型集成到自己的应用程序中进行实际应用。 总之,yolov4环境搭建pytorch(gpu)需要注意的一些事项,包括安装Anaconda、创建Python虚拟环境、下载yolov4-tiny的模型权重、编写实现代码、使用自己的数据集进行模型训练和使用模型进行目标检测。只有完成以上步骤,我们才能在本地或服务器上更好地应用yolov4算法。
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值