利用腾讯云GPU服务器一键化部署 进行yolov5模型训练

购买服务器:

首先购买云服务器,我选的是按量计费的。

利用腾讯云GPU服务器一键化部署 进行yolov5模型训练_第1张图片

镜像我选择的ubuntu20.04,已经配置cuDNN和CUDA,省去手动配置的麻烦。

利用腾讯云GPU服务器一键化部署 进行yolov5模型训练_第2张图片

 等待cuda,cuDNN安装完成:

利用腾讯云GPU服务器一键化部署 进行yolov5模型训练_第3张图片

安装环境:

运行以下指令安装anaconda:

# 安装anaconda
wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-2023.03-1-Linux-x86_64.sh # 下载anaconda
chmod +x Anaconda3-2023.03-1-Linux-x86_64.sh
./Anaconda3-2023.03-1-Linux-x86_64.sh

下载的时间很长,下载过程中就可以上传文件了

下载完成后运行

运行过程中不断会有协议跳出来,一直按enter一直到最后输入 yes

利用腾讯云GPU服务器一键化部署 进行yolov5模型训练_第4张图片

利用腾讯云GPU服务器一键化部署 进行yolov5模型训练_第5张图片

 输入完yes之后会让你确认路径,直接enter就好。

利用腾讯云GPU服务器一键化部署 进行yolov5模型训练_第6张图片

 问你要不要初始化,再次输入yes

利用腾讯云GPU服务器一键化部署 进行yolov5模型训练_第7张图片

安装完后还不能用,会出现:

利用腾讯云GPU服务器一键化部署 进行yolov5模型训练_第8张图片

 输入:

vim ~/.bashrc

 来到行末,添加:

export PATH=$PATH:/home/ubuntu/anaconda3/bin

利用腾讯云GPU服务器一键化部署 进行yolov5模型训练_第9张图片

之后保存退出。

执行

source .bashrc

成功:

conda换源

输入:

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.ustc.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/bioconda/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/menpo/
conda config --set show_channel_urls yes
vim .condarc

 会出现:

利用腾讯云GPU服务器一键化部署 进行yolov5模型训练_第10张图片

删除倒数第二行defauls,保存退出。

   创建yolov5环境:

conda create -n yolov_5 python=3.8

 过程中遇到yes/no就输yes,或者y/n就输入y

激活yolov5环境并安装相关依赖:

conda activate yolov_5
pip install -r yolov5/requirements.txt

在安装依赖期间可以上传文件。

上传文件

地址:WinSCP :: Official Site :: Download利用腾讯云GPU服务器一键化部署 进行yolov5模型训练_第11张图片

打开,连接服务器:

利用腾讯云GPU服务器一键化部署 进行yolov5模型训练_第12张图片

 密钥提示点击[是]

之后选中文件夹,点击上传,建议yolov5和yolov5s.pt文件夹和文件自己上传,使用git速度慢易出错

利用腾讯云GPU服务器一键化部署 进行yolov5模型训练_第13张图片

  等待文件上传完成和依赖安装完成

开始训练

unzip 你的文件名.zip

修改nc数量:

vim models/yolov5s.yaml

接下来就可以直接训练了

python train.py --img 640 --batch 32 --batch_size 64 --epochs 300 --save_period 10 --data ../down/down.yaml --weights yolov5s.pt --cfg ./models/yolov5s.yaml --name down --nosave --cache

你可能感兴趣的:(腾讯云,gpu,YOLO)