yolov8的深度学习环境安装(cuda12.4、ubuntu22.04)
目录
一、先安装基础环境包
1.首先给Ubuntu安装Chrome浏览器(搜索引擎换成百度即可)
2、ubuntu 22.04中文输入法安装
3、安装 terminator
4、安装WPS for Linux
5、安装其它之前需要先安装anaconda
6、安装配置anaconda
7、安装完成anaconda后创建一个虚拟环境
8、安装pip:
报错:无法获取 dpkg 前端锁 (/var/lib/dpkg/lock-frontend),是否有其他进程正占用它?
解决:
2.更新 APT 包列表:
9、小鱼的大佬的一键配置
10、ubuntu安装wps
11、安装netron
12、安装向日葵
13、安装labelimg
二、安装pytorch前首先安装显卡驱动
1、输入查看驱动
三、安装配置pytorch
官网:pytorch下载地址
1.首先查看查看显卡使用情况
2、我的准备装cuda12.4
Linux and Windows
四、接下来安装cuda、cudnn
官网:cuda官网
1、选择自己版本
2、这里我们选择runfile(local)的安装方式
3、安装cuda
五、安装cudnn
安装cudnn
这里直接参考了我的另一篇文章:Ubuntu20.04配置深度学习环境yolov5最简流程_command 'nvidia-smi' not found, but can be install-CSDN博客
python convert.py ../model/yolov8n.onnx rk3588
一、先安装基础环境包
1.首先给Ubuntu安装Chrome浏览器(搜索引擎换成百度即可)
安装命令:打开终端直接输入
wget https://dl.google.com/linux/direct/google-chrome-stable_current_amd64.deb
sudo dpkg -i google-chrome-stable_current_amd64.deb
sudo apt-get -f install
安装环境包经常使用放在这里
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
2、ubuntu 22.04中文输入法安装
sudo apt-get install fcitx-googlepinyin
3、安装 terminator
sudo add-apt-repository ppa:gnome-terminator
sudoapt-get update
sudo apt install terminator
4、安装WPS for Linux
进入搜狗for linux官网下载搜狗输入法 ,下载x86版本
WPS for linux
5、安装其它之前需要先安装anaconda
conda下载地址
-
先去官网下载好anaconda后面是.sh文件
-
输入命令(命令上对应你自己下载的版本号):
sh Anaconda3-2021.11-Linux-x86_64.sh
将Anconda3的安装地址添加进PATH即可。首先打开~/.bashrc,然后在将
export PATH=$PATH:/home/sxj/anaconda3/bin
3.添加到末尾,最后更新一下~/.bashrc。依次执行下面代码。
sudo gedit ~/.bashrc
source ~/.bashrc
添加内容【注意这里的export PATH=$PATH:/home/sxj/anaconda3/bin需要切换成自己的用户名】:
完成之后测试:
conda env list
6、安装配置anaconda
anaconda添加国内镜像源
#添加镜像源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
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/r
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
#显示镜像通道
conda config --show channels
7、安装完成anaconda后创建一个虚拟环境
conda create -n yolo8 python=3.10
查看自己的虚拟环境
conda env list
激活虚拟环境
conda activate yolo8
接下来需要在虚拟环境里面安装pytorch
8、安装pip:
sudo apt-get install python3-pip
报错:无法获取 dpkg 前端锁 (/var/lib/dpkg/lock-frontend),是否有其他进程正占用它?
解决:
1.清除 APT 缓存:
sudo rm /var/lib/apt/lists/lock
sudo rm /var/cache/apt/archives/lock
sudo rm /var/lib/dpkg/lock-frontend
sudo rm /var/lib/dpkg/lock
2.更新 APT 包列表:
sudo apt-get update
sudo apt-get install python3-pip
9、小鱼的大佬的一键配置
打开ubuntu终端,输入:
wget http://fishros.com/install -O fishros && . fishros
10、ubuntu安装wps
执行:
sudo dpkg -i wps-office*.deb
11、安装netron
pip3 install netron -i https://pypi.tuna.tsinghua.edu.cn/simple
netron best.onnx
netron best.pt
12、安装向日葵
向日葵官网下载地址:
下载地址
sudo dpkg -i SunloginClient_15.2.0.63064_amd64.deb
sudo apt --fix-broken install
13、安装labelimg
pip install labelimg -i https://pypi.tuna.tsinghua.edu.cn/simple
启用:
labelImg
14、安装百度网盘
sudo dpkg -i baidunetdisk_4.17.7_amd64.deb
二、安装pytorch前首先安装显卡驱动
1、输入查看驱动
nvidia-smi
提示Command 'nvidia-smi' not found, but can be installed with:说明你还没有安装显卡驱动
选择适合自己的
sudo apt-get install nvidia-driver-535
设置完密码reboot就可以了
这时如果弹出没有可用的附加驱动或附加驱动为空
解决方法:
sudo add-apt-repository ppa:graphics-drivers/ppa
软件和更新即可显示驱动
完成后重启在输入nvidia-smi就可以看到电脑可以安装cuda的最高版本号
三、安装配置pytorch
官网:pytorch下载地址
1.首先查看查看显卡使用情况
打开终端输入:
nvidia-smi
查看可以安装cuda的版本(低于这个或等于这个版本都可以安装)
安装前先进入自己环境:
conda activate yolo8
2、我的准备装cuda12.4对应版本的pytorch
conda install pytorch==2.5.0 torchvision==0.20.0 torchaudio==2.5.0 pytorch-cuda=12.4 -c pytorch -c nvidia
如果这种方式安装太慢!!!
那么你需要在cuda及cudnn都安装完成后,在执行如下命令:
不要现在执行,否则可能会和你的cuda及cudnn版本有冲突
pip3 install torch==2.5.0 torchvision==0.20.0 -i https://pypi.mirrors.ustc.edu.cn/simple/
或在cuda及cudnn都安装完成后让其自己选择版本
pip3 install torch torchvision -i https://pypi.mirrors.ustc.edu.cn/simple/
pip install torch==2.5.0 torchvision==0.20.0 torchaudio==2.5.0 --index-url https://download.pytorch.org/whl/cu124
其它的找适合自己版本就行
Linux and Windows
# CUDA 11.8 conda install pytorch==2.5.0 torchvision==0.20.0 torchaudio==2.5.0 pytorch-cuda=11.8 -c pytorch -c nvidia # CUDA 12.1 conda install pytorch==2.5.0 torchvision==0.20.0 torchaudio==2.5.0 pytorch-cuda=12.1 -c pytorch -c nvidia # CUDA 12.4 conda install pytorch==2.5.0 torchvision==0.20.0 torchaudio==2.5.0 pytorch-cuda=12.4 -c pytorch -c nvidia # CPU Only conda install pytorch==2.5.0 torchvision==0.20.0 torchaudio==2.5.0 cpuonly -c pytorch
国内镜像源经常用放在这里
-i https://pypi.tuna.tsinghua.edu.cn/simple
四、接下来安装cuda、cudnn
安装cudnn官网(参考文章Ubuntu安装cuda与cudnn,亲测可用)
官网:cuda官网
1、选择自己版本
我的是cuda12.4
2、这里我们选择runfile(local)
的安装方式
执行:
wget https://developer.download.nvidia.com/compute/cuda/12.4.0/local_installers/cuda_12.4.0_550.54.14_linux.run
sudo sh cuda_12.4.0_550.54.14_linux.run
3、安装cuda
在运行sudo sh cuda_11.1.0_455.23.05_linux.run
时,不要安装驱动、不要安装驱动、不要安装驱动,重要的事情说三遍
操作为:选择continue
->输入accept
->按空格
取消Driver
选项->选择Install
安装
完成安装
4、安装完成后需要配置环境变量:
sudo gedit ~/.bashrc
在末尾添加:
export PATH=$PATH:/usr/local/cuda-12.4/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-12.4/lib64
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda-12.4/lib64
5、最后;使更改生效:
source ~/.bashrc
验证环境变量是否正确设置:
6、验证环境变量是否正确设置:
echo $PATH
echo $LD_LIBRARY_PATH
echo $LIBRARY_PATH
确保输出中包含 /usr/local/cuda-12.4/bin 和 /usr/local/cuda-12.4/lib64
7、验证 CUDA 和驱动程序是否正常工作:
nvidia-smi
五、安装cudnn
安装cudnn
官网下载cuda
对应版本的cudnn
:cuDNN Archive | NVIDIA Developer
这里我们选择的是8.8.1版本的cudnn
下载一下1个文件:
下载完成:
之前是三个dep文件,最新的版本没有了
tar -xf cudnn-linux-x86_64-8.8.1.3_cuda12-archive.tar.xz
解压完成:
1.创建必要的目录:
sudo mkdir -p /usr/local/cuda/include
sudo mkdir -p /usr/local/cuda/lib64
复制文件:
2.将 include 和 lib 目录中的文件复制到 /usr/local/cuda 目录下:
1.进入到解压后的文件夹下:
cd /home/sxj/anzhuang/cudnn-linux-x86_64-8.8.1.3_cuda12-archive
sudo cp -r include/* /usr/local/cuda/include/
sudo cp -r lib/* /usr/local/cuda/lib64/
3.更新环境变量:
编辑 ~/.bashrc 文件,添加以下行以更新环境变量:
gedit ~/.bashrc
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
export PATH=/usr/local/cuda/bin:$PATH
然后使更改生效:
source ~/.bashrc
4.验证 cuDNN 是否正确安装:
dpkg -l | grep cudnn
检查 CUDA 和 cuDNN 版本:
nvcc --version
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
python
import torch
print(torch.cuda.is_available())
以下命令来检查 CUDA 版本:
nvcc -V
直接参考我的另一篇文章:yolov8训练pt模型转换为rknn模型_部署在RK3588上--整个流程-CSDN博客
安装yolov8环境
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
运行:
python3 train.py
最后测试了yolov8成功完成gpu调用
解决字体报错:Arial.tff字体,YOLO中下载。已解决_arial.ttf字体-CSDN博客