训练环境配置的话就仍然以YOLOv8为例,来说明如何配置深度学习训练环境。这部分内容比较简单,主要是安装miniAnaconda以及安装torch和torchvision.
首先是miniAnaconda的安装(参考官网的教程Miniconda — Anaconda ),执行以下指令即可:
- mkdir -p ~/miniconda3
- wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O ~/miniconda3/miniconda.sh
- bash ~/miniconda3/miniconda.sh -b -u -p ~/miniconda3
- rm -rf ~/miniconda3/miniconda.sh
然后初始化一下,重新打开后就直接是anaconda环境了.
~/miniconda3/bin/conda init bash
最后利用anaconda创建YOLOv8训练环境,这里使用的环境是python-3.8,torch-1.13,torchvision-0.14,cuda版本也和WSL的cuda版本保持