栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

Ubuntu18.04+yoloV5环境搭建

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

Ubuntu18.04+yoloV5环境搭建

注意事项:一定要找对应匹配的版本安装,否则那就是教训!*

一、查看自己服务器的显卡
  1. 查询命令 nvidia-smi -L

    显示有2个显卡,第一个720的算力低,不支持pytorch1.6,最高支持1.2,但是yoloV5要求1.6以上,所以在使用过程中禁用掉第一个。
    下来所有的安装都以3070为准。
二、安装驱动
  1. 下载显卡驱动

    显卡驱动地址

  2. 查看是否安装成功
    命令:nvidia-smi

    驱动的版本是470.57;最高支持的CUDA版本是11.4版本。

三、安装CUDACUDNN
  1. 下载CUDACUDNN
    CUDA下载地址
    CUDNN下载地址
    由于外网下载比较慢。分享我下载的一些版:

    链接:https://pan.baidu.com/s/1RluZ3hX3I_Q5yH8BpfHtYA
    提取码:nabv

  2. 安装
    将下载好的文件放到指定位置,具体参考这个博文.详细安装过程
四、安装Anaconda

具体安装参考:Anaconda+Pycharm

五、安装pytorch
  1. 下载
    pytorch下载连接
    注意:由于我的CUDA11.1,所以我的pytorch用的是pytorch1.8.0

  2. 安装

    1)首先通过conda activate进入conda终端;
    2)创建虚拟环境 conda create -n py1 python=3.7
    3)进入虚拟环境conda activate py1
    4)找到你下载的pytorch的文件,直接安装

六、下载yolov5,安装对应的库**
  1. yolov5下载

    yolov5下载地址

  2. 下载后解压,在打开终端,进入该文件夹,输入以下命令配置环境:pip install -r requirement.txt

  3. 若安装完,还有一些没安装完,则直接pip安装
    若安装torchvirsion时候,一定注意版本对应。

七、参考文件:

windows安装教程,详细

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/461340.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号