yolov5体验

无须安装CUDA，只需要有NVIDIA图形驱动即可

1. 安装Miniconda

miniconda下载地址

1.1 安装细节

• 一个对勾都不要选择
  

1.2 配置环境变量

在环境变量Path中添加如下变量

C:\Server\miniconda
C:\Server\miniconda\Scripts
C:\Server\miniconda\Library\bin
1
2
3

2. 创建虚拟环境

2.1 创建虚拟环境yolov5

conda create -n yolov5
1

2.2 进入虚拟环境

conda activate yolov5
1

• 若出现如下错误

CommandNotFoundError: Your shell has not been properly configured to use 'conda activate'.
To initialize your shell, run
 
    $ conda init <SHELL_NAME>
 
Currently supported shells are:
  - bash
  - fish
  - tcsh
  - xonsh
  - zsh
  - powershell
 
See 'conda init --help' for more information and options.
 
IMPORTANT: You may need to close and restart your shell after running 'conda init
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16

Windows：执行如下命令后即可使用命令conda activate yolov5

conda init cmd.exe
1

Linux：执行如下命令后即可使用命令conda activate yolov5

conda init bash
1

2.3 更换清华镜像源

清华镜像网站

3. PyTorch安装

3.1 进入pytorch官网下载v1.8.2

• 本人显卡为1650,故选择CUDA 10.2版本执行命令

3.2 网速过慢

1. 使用迅雷下载文件
   https://download.pytorch.org/whl/lts/1.8/cu102/torch-1.8.2%2Bcu102-cp38-cp38-win_amd64.whl
2. 使用pip安装

pip install  C:\torch-1.8.2+cu102-cp38-cp38-win_amd64.whl
1

3. 执行3.1中命令（本人采用CUDA版本为10.2）

pip3 install torch==1.8.2 torchvision==0.9.2 torchaudio==0.8.2 --extra-index-url https://download.pytorch.org/whl/lts/1.8/cu102
1

4. yolov5源码下载

4.1 github下载source

https://github.com/ultralytics/yolov5/releases/v7.0/
1

4.2 下载

• 使用迅雷下载红框选中的源码下载即可
• 解压至C:
  

4.3 修改requirments.txt

1. 注释掉torch和torchvision，若不注释，会使用CPU
2. numpy版本号更改1.20.3
3. Pillow版本号更改为5.3.0

# YOLOv5 
声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/2023面试高手/article/detail/218715