UBUNTU 18.04 GPU + miniconda3 + pytorch + yolo v5 安装测试

安装CUDA10.0

检查系统是否满足要求

有支持CUDA的GPU  lspci | grep -i nvidia
支持CUDA的操作系统。（Ubuntu没有问题）
已经安装了合适版本的gcc（CUDA10.0要求7.3.0）
系统有正确的Kernel Header  执行 sudo apt install linux-headers-$(uname -r) 即可
已经卸载了有冲突的CUDA版本

下载CUDA10.0安装文件https://developer.nvidia.com/cuda-10.0-download-archive 选择Linux、X86、Ubuntu18.04、runfile下载
禁用nouveau驱动：

sudo vi /etc/modprobe.d/blacklist-nouveau.conf，在文件中编辑：blacklist nouveauoptions nouveau modeset=0sudo update-initramfs -u
重启机器

删除已有驱动

sudo apt purge nvidia-*

安装驱动

sudo ubuntu-drivers autoinstall （在此之前可以执行命令 ubuntu-drivers devices查看系统推荐的驱动程序，也可以在桌面上执行【软件和更新】，在附加驱动中选择推荐的驱动）

重启计算机
执行安装命令

sudo sh cuda_10.0.130_410.48_linux.run，注意一定不要选择安装驱动，其余回答yes或默认值

设置环境变量：vi ~/.bashrc，编辑如下内容：

export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

保存退出，执行 source ~/.bashrc
验证安装是否成功

cd /usr/local/cuda/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
显示 Result=PASS以及CUDA驱动信息，表示安装成功

miniconda

前言
conda是一种通用包管理系统，旨在构建和管理任何语言和任何类型的软件。举个例子：包管理与
pip的使用类似，环境管理则允许用户方便地安装不同版本的python并可以快速切换。

Anaconda则是一个打包的集合，里面预装好了conda、某个版本的python、众多packages、科学
计算工具等等，就是把很多常用的不常用的库都给你装好了。

Miniconda，顾名思义，它只包含最基本的内容——python与conda，以及相关的必须依赖项，对
于空间要求严格的用户，Miniconda是一种选择。就只包含最基本的东西，其他的库得自己装。
Miniconda会比Anaconda小很多，Miniconda足够用，两者安装都很简单。

安装miniconda

mkdir miniconda
cd miniconda
wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda/Miniconda3-latest-Linux-x86_64.sh

执行程序

bash Miniconda3-latest-Linux-x86_64.sh

这里会弹出一个软件协议条款让你阅读，这时候直接按下Ctrl+C便可以跳过阅读过程，直接按照提示，输入yes（有yes/no的都选yes），然后按回车键，同意软件协议条款。接下来继续按回车，将miniconda安装到电脑上。安装好之后，提示如下：

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-EzrqUqDD-1649832223958)(/storage/2022/04-13/4sLYRZey9EFykCPp4eqflYGf3zLlI38ioO1Syfqp.png)]

重启终端之后，就可以使用miniconda了。但先换源，将conda和pip的软件源修改成清华的源，下载安装软件会快很多：

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
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

换完源再次重启终端之后，你会发现每次打开终端默认都会自动进入到miniconda的base环境中，终端中多了“base”字样。这样会拖慢终端打开速度，并且有可能干扰到其它软件的安装。要退出的话，必须每次打开终端之后先执行conda deactivate命令，让人很难受。执行如下命令，便可以解决终端每次打开都进入conda的base环境的问题：

conda config --set auto_activate_base false

至此 miniconda 已安装完成

安装labelimg 标注软件

输入命令获取labelimg 工程

git clone https://github.com/Ruolingdeng/labelImg.git

然后打开安装包labelImg，并右键打开终端或快捷键Ctrl+Alt+T打开终端并cd 目录到labelImg下，输入

sudo apt-get install pyqt5-dev-tools #安装过实训系统的电脑应该不需要这一步，因为已经装过了

安装完之后，打开安装包labelImg，并右键打开终端，然后输入：

sudo pip3 install -r requirements/requirements-linux-python3.txt
make qt5py3

然后运行软件

python3 labelImg.py

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-Iisu9jvK-1649832223959)(/storage/2022/04-13/XYogpN8YL8imakrrlYeLauOShr7pIZN0N7ZnBmWV.png)]

出现该界面为软件安装成功

Yolov5

一.搭建YOLOv5深度学习环境

conda create -n yolov5 python=3.10  //yolov5是虚拟环境的名字
source activate yolov5

此时，已进入 miniconda的虚拟环境

二、下载YOLO V5 工程

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

进入yolo v5 目录下

cd yolov5

安装 yolo v5 依赖

pip install -r requirements.txt

使用官网命令安装pytorch

pip install torch==1.8.0+cpu torchvision==0.9.0+cpu torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html

至此 Yolo v5 环境已搭建完成

可以使用测试程序对进行简单图片识别测试

python detect.py --source ./data/images/