YOLOV5 配 pytorch 环境

YOLOV5 配 pytorch 环境

以下操作建立在本人已安装cuda 10.2

1. 安装pytorch

(1) 安装Anaconda

1)先去官方地址下载好对应的安装包 下载地址:

https://www.anaconda.com/download/#linux

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 项目

git clone https://github.com/ultralytics/yolov5.git

(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.0torchvision>=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

测试成功,证明环境安装成功

 

  • 0
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值