ubuntu18.04搭建yolov5环境

  1. 安装minianaconda
  2. 创建虚拟环境
    image.png

    conda create -n yolov5-torch python=3.8
  3. 激活与退出环境

    conda activate yolov5-torch
    conda deactivate
  4. 安装pytorch
  5. 选择什么版本的cuda?
    torch官网:https://pytorch.org/
    image.png
# 查看显卡驱动对应的cuda:
nvidia-smi
# 查询结果为:11.7

# 查看runtime对应的cuda
nvcc --version
# 查询结果:没有安装!!!
  1. 安装nvcc,实际就是安装cuda
    cuda下载地址:https://developer.nvidia.com/...
    cuda版本选择:不得高于显卡驱动版本对应的cuda版本
    原因:https://stackoverflow.com/que...
    image.png
    选择 runfile版本,不会替换掉之前的高版本驱动
sudo bash cuda_11.7.0_515.43.04_linux.run
# 在这里一直失败,提示没有安装驱动

#卸载之前的驱动
sudo service lightdm status #关闭X服务
sudo service lightdm stop
sudo service lightdm status
sudo apt-get --purge remove nvidia*
sudo apt-get autoremove nvidia*

sudo bash cuda_11.7.0_515.43.04_linux.run
# 仍然失败

sudo reboot
# 重复之前的步骤成功

~/.bashrc
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

#验证
source ~/.bashrc
nvcc -V
nvidia-smi

# 安装多版本cuda,比如安装cuda11.6
sudo bash cuda_11.6.0_510.39.01_linux.run 
# 安装过程中去掉驱动项即可,其他照常安装
# 安装完成后 /usr/local 出现 cuda-11.6/

#切换不同版本cuda
cd /usr/local
sudo rm -rf cuda
sudo ln -s /usr/local/cuda-11.6 /usr/local/cuda

# 验证
nvcc -V
  1. 安装torch

    conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
  2. 下载yolov5源码

    git clone https://github.com/ultralytics/yolov5.git
  3. 下载权重到 yolov5/weights文件夹下,验证推理结果

    mkdir weights
    cd weights
    wget https://github.com/ultralytics/yolov5/releases/download/v6.2/yolov5s.pt
    python detect.py --weights weights/yolov5s.pt

    权重下载地址:https://github.com/ultralytic...

  4. 训练自己的数据集
    todo
1 声望
0 粉丝
0 条评论
推荐阅读
ubuntu系统笔记本电池坏掉的情况下,性能严重下降的解决办法
问题描述开机出现以下问题 {代码...} 开机选择diagnose进行检查发现 {代码...} 显示电池已经到达生命尽头问题原因分析 电池坏掉后cpu动不动就跑满了,因为此时cpu的工作模式被修改了安装cpufrequency工具 {代码.....

frank阅读 166

从头开始进行CUDA编程:Numba并行编程的基本概念
GPU(图形处理单元)最初是为计算机图形开发的,但是现在它们几乎在所有需要高计算吞吐量的领域无处不在。这一发展是由GPGPU(通用GPU)接口的开发实现的,它允许我们使用GPU进行通用计算编程。这些接口中最常见的...

deephub阅读 634

封面图
通过pip安装cuda
五六年前深度学习还是个新鲜事的时候,linux下显卡驱动、CUDA的很容易把小白折磨的非常痛苦,以至于当时还有一个叫manjaro的发行版,因为驱动安装简单流行。老黄也意识到了这个问题,增加了很多新的安装方式。

cyoahs阅读 478

ubuntu18.04 配置docker内的gui程序 显示到宿主机器
使用场景描述本机系统:ubuntu18.04 本机显卡:nvidia,已经安装了显卡驱动服务器系统:ubuntu18.04container运行位置:服务器gui显示位置:本机系统

frank阅读 364

ubuntu18.04安装nvidia显卡驱动
查看本机显卡型号 {代码...} 去nvidia官网下载对应型号的驱动nvidia驱动下载改变驱动执行权限 {代码...} 关闭电脑lightdm驱动 {代码...} ctrl+alt+f1安装 {代码...} 默认执行即可重新回到界面 {代码...} 验证是否...

frank阅读 314

ubuntu18.04安装miniconda
conda官网地址conda下载地址安装过程记录 {代码...} 安装路径选择:/opt/miniconda3验证是否安装成功 {代码...} 在~/.bashrc中添加环境变量后可以使用

frank阅读 294

使用Opencv4和YOLOv4(XTDrone)训练模型遇到问题的记录(二)
使用Opencv4和YOLOv4(XTDrone)训练模型遇到问题的记录(二)Written By PiscesAlpaca(双鱼座羊驼) [TOC] 一、Opencv4安装问题记录1.在cmake时,需要手动标记生成pgk-config文件命令如下: {代码......

双鱼座羊驼阅读 287

1 声望
0 粉丝
宣传栏