目录
1. 查看电脑状况
深度学习的训练对于电脑显卡要求较高,若电脑没有独立显卡(NVIDIA卡)或者是AMD的显卡无法使用GPU进行训练,只可以用CPU进行训练,会慢很多。若不清楚电脑有无显卡可以打开任务管理器,点击性能往下找GPU,就可以看到是否有独立显卡,不清楚是否为独立显卡则搜索型号。
若有独立显卡则安装GPU版本的pytorch,若无独立显卡则安装CPU版本的pytorch。
点击下载训练源码 夸克网盘下载 ,建议先全部转存提前下载,若有需要下载的资源失效,可至公众号获取百度盘链接下载。
YOLOv9网络结构图,论文必备,无水印图可 微信公众号-笑脸惹桃花 回复“999” 获取。
2. 安装所需软件
推荐安装Anaconda3+Pycharm,会安装或者安装过了则跳过这一步
2.1 Anaconda3安装
Anaconda3由于是国外网站下载较慢,推荐通过清华镜像源安装
选择带有Anaconda3...Windows...exe 字样的进行下载,建议从最新版往后边找几个,最新版一般不够稳定。下载完之后打开Anaconda3进行安装,一直点下一步,选Just Me,安装路径可以直接复制粘贴修改到 D:\Anaconda3 ,也可以修改到其他路径,也可以点击网盘下载。
这里建议全选。
2.2 Pycharm安装
安装Pycharm可以直接去官网下载,速度较快,也可以点击网盘下载。
往下拉下载第二个免费版就可以。下载完之后安装,一直点下一步,遇到选择路径 修改路径到D盘或者其它除C盘,再遇到勾选的选项全部勾选就可以。
3. 创建虚拟环境,安装pytorch
3.1 创建虚拟环境
按下Win键,输入anaconda prompt,打开推荐的应用
默认进入的是base环境,此时需要新建环境,在新建环境之前建议更改默认的pip源和conda源。
更改代码如下,直接输入即可,这里选择更换的是中科大源
conda config --remove-key channels
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.bfsu.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes
pip config set global.index-url https://mirrors.ustc.edu.cn/pypi/web/simple
此时新建虚拟环境(需要关闭加速软件),这里创建一个名为yolov9,python版本为3.9的虚拟环境,也可以修改为其他名。
conda create -n yolov9 python=3.9
注意:此时如果报错
UnavailableInvalidChannel: HTTP 404 NOT FOUND for channel anaconda/pkgs/free <https://mirrors.ustc.edu.cn/anaconda/pkgs/free>
The channel is not accessible or is invalid.
You will need to adjust your conda configuration to proceed.
Use `conda config --show channels` to view your configuration's current state,
and use `conda config --show-sources` to view config file locations.
等错误,需要修改.condarc文件的内容,具体路径为:c:\users\用户名\.condarc(找不到请显示隐藏文件)
使用记事本打开,复制下面这一段进去替换掉原本所有的内容,替换完之后保存即可
channels:
- defaults
show_channel_urls: true
channel_alias: http://mirrors.tuna.tsinghua.edu.cn/anaconda
default_channels:
- http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
- http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
- http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
- http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro
- http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
custom_channels:
conda-forge: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
msys2: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
bioconda: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
menpo: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
pytorch: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
simpleitk: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
此时再次创建yolov9虚拟环境
conda create -n yolov9 python=3.9
显示如下则正常创建,输入y 并回车
创建完之后输入
conda activate yolov9
进入yolov9环境之中,此时有NVIDIA显卡安装GPU版看3.2,其它需要安装CPU版看3.3。
3.2 安装cuda,pytorch
3.2.1 安装cuda
在安装pytorch前需要安装cuda,(若无显卡则跳过)下载cuda前需要先查看显卡支持的CUDA版本最高是多少,按下win+r键,输入cmd,在打开的页面输入:nvidia-smi ,即可查看
上图红框位置显示即为cuda最高支持版本,本教程最高版本达到11.8即可,若没有达到则需要更新显卡驱动。
CUDA Toolkit Archive | NVIDIA Developerhttps://developer.nvidia.com/cuda-toolkit-archive
下载完之后运行安装,建议安装到默认路径,所以C盘需要留有20G以上的存储空间,一直点击下一步,直到出现这个界面,不要点击精简,选择自定义,然后全部勾选再一直下一步即可。
安装完成后可以再次在cmd里输入命令:nvcc -V 查看,如下显示即安装成功
安装cudnn
进入cudnn官网,选择合适版本的文件。
cuDNN Archive | NVIDIA Developerhttps://developer.nvidia.com/rdp/cudnn-archive进入后在文件列表中选择cudnn版本与上面cuda安装相匹配的版本。
下载Windows版本的压缩包文件。
下载需要登录,也可以复制下载链接打开迅雷下载,或者点击链接下载下载cudnn。
将得到的压缩文件进行解压,解压后得到下图三个文件夹,全选复制进cuda的文件夹中进行覆盖替换,替换完成后即cudnn安装完成。按照本文教程安装的cuda的文件夹默认在 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8 目录下。
3.2.2 安装pytorch(GPU版)
打开prompt 输入conda activate yolov9进入yolov9环境,之后输入即可安装pytorch,耐心等待安装完成(需要注意关闭加速软件否则会下载失败)
pip install torch==2.0.0+cu118 torchvision==0.15.1+cu118 --extra-index-url https://download.pytorch.org/whl/cu118
torch及相关库比较大,需要耐心等待下载完之后出现 done 则安装完成,因为是外网,如果下载不下来,可以点此下载torch 夸克网盘下载,然后通过pip install安装本地文件的方式安装,可以参考下图的方式,将torch放入d盘的test文件夹下,通过cd将文件目录跳转,再输入如下指令。(只支持python3.9版本,其余版本无法安装)
pip install torch-2.0.0+cu118-cp39-cp39-win_amd64.whl
耐心等待安装完成后,再输入下面的指令通过pip安装其余库。
pip install torch==2.0.0+cu118 torchvision==0.15.1+cu118 --extra-index-url https://download.pytorch.org/whl/cu118
安装完成如上图所示
3.3 安装CPU版本
有显卡的按照上述流程安装后即可跳过本部分,无显卡则需要按照下述方法进行,
打开prompt ,进入yolov9环境之后,输入如下命令即可安装cpu版本的pytorch
pip install torch==2.0.0 torchvision==0.15.1 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cpu
安装完成则可以进行下一步骤。
4. 配置yolov9环境
4.1 下载yolov9源码
yolov9源码地址:
点进去直接下载代码,找到点击code-> download zip 下载代码。进不去可能需要科学上网,或者点击网盘下载(压缩包内附带yolov9-t-converted.pt和yolov9-s-converted.pt)。
下载完成之后解压到D盘或其它容量大的盘文件夹内
4.2 安装yolov9所需库
需要通过anaconda prompt 进入yolov9环境中,按照下图cd到yolov9根目录下,这里替换为自己的文件夹
进入yolov9根目录之后,需要通过pip安装所需库,指令如下
pip install -r requirements.txt
耐心等待全部安装完成即可
至此环境安装完成
4.3 pycharm打开yolov9
点击鼠标右键yolov9文件夹通过pycharm打开,打开后需要配置虚拟环境,点击file-settings,点击project:yolov9-main,点击python interpreter,点击右边add....,如果是中文则翻译后点击相应含义的位置。
点击Virtualenv Environment - Existing,点击右边三个点,找到刚才设置的D:\Anaconda3\envs\yolov9\python.exe 路径,之后一直点确定即可,耐心等待环境加载完毕。
再次进入yolov9源码页面,往下翻找到权重文件,或者 点击网盘下载。
点击蓝字下载,可以选择yolov9-t-converted.pt或者yolov9-s-converted.pt,适合性能较低的电脑且效果还可以
5. 验证环境
下载完成后将权重文件复制到yolov9根目录下,此时可以去网上下载一只狗狗的jpg图片,将其放到data/images目录下。
检测环境是否有问题可以运行detect.py,修改方框中为刚下载的权重文件名即可
输出下图即环境正常
可以顺着路径查看检测后的图片,可爱的阿拉斯加一张~
报错 ImportError: DLL load failed while importing _imaging: 找不到指定的模块,是pillow版本问题导致的错误,输入如下指令降低pillow的版本即可解决此问题。
pip install pillow==8.4.0
如果报错中有numpy相关的语句,如 RuntimeError: Numpy is not available ;compiled using NumPy 1.x cannot be run in NumPy 2.1.2 as it may crash. To support both 1.x and 2.x versions of NumPy, modules must be compiled with NumPy 2.0. 等等,可以将numpy降级,输入如下代码即可解决此类问题。
pip install numpy==1.26.3
YOLOv9训练自己的数据集可以点击下面的链接进行训练。