YOLOV5 配 pytorch 环境
以下操作建立在本人已安装cuda 10.2
1. 安装pytorch
(1) 安装Anaconda
1)先去官方地址下载好对应的安装包 下载地址:
2)然后安装anaconda
bash ~/Downloads/Anaconda3-2020.11-Linux-x86_64.sh
需要添加环境变量。 编辑~/.bashrc 文件
vi ~/.bashrc
在最后一行加上
export PATH=/home/bai/anaconda3/bin:$PATH
根据自己的名字修改用户名 bai,然后更新环境变量
source ~/.bashrc
3)添加Aanaconda国内镜像配置
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
(2) 安装pytorch
注意:需要安装pytorch 1.6以上的版本, 首先为pytorch创建一个anaconda虚拟环境,环境名字可自己确定,这里本人使用pytorch1.7作为环境名:
conda create -n pytorch1.7 python=3.7
注:现有安装包是torch-1.7.1, torchvision-0.8.2, 适配python3.7
安装成功后激活pytorch1.7环境:
conda activate pytorch1.7
因执行命令下载很慢,下面是百度网盘torch 和torchvision 的安装包连接
链接:https://pan.baidu.com/s/16Wkhv6Er73eduZciKN3b2w
提取码:0mv0
cd 到安装包目录下运行:
pip install torch-1.7.1+cu101-cp37-cp37m-linux_x86_64.whl
pip install torchaudio-0.7.2-cp37-cp37m-linux_x86_64.whl
pip install torchvision-0.8.2+cu101-cp37-cp37m-linux_x86_64.whl
2 yolov5 项目克隆和安装
(1)克隆yolov5 项目
(2)安装所需库
使用清华镜像源, 在yolov5目录下执行:
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple -r requirements.txt
注: 该方法下载慢,可打开改文件,逐个下载:
pip install Cython -i https://pypi.tuna.tsinghua.edu.cn/simple
其中,torch>=1.6.0,torchvision>=0.7.0,已经在步骤(2)中已经安装,不需要再装。
(3)下载预训练权重文件
下载yolov5s.pt,yolov5m.pt,yolov5l.pt,yolov5x.pt权重文件,并放置在weights文件夹下
百度网盘下载链接:
链接:https://pan.baidu.com/s/1hjDQIfPlqOmL4HXoi8Ja1g
提取码:wexh
(4)安装测试
测试图片:
CUDA_VISIBLE_DEVICES='3' python detect.py --source ./inference/images/ --weights weights/yolov5s.pt --conf 0.4
测试成功,证明环境安装成功