【服务器使用】曙光Gridview集群运行YOLO(pytorch)项目实战

1.构建作业脚本-yolov9.sh

#!/bin/bash
#SBATCH -J yolov9              # 作业名
#SBATCH -N 1                         #请求节点数
#SBATCH -n 16                          #请求核心数
#SBATCH --gres=gpu:1
#SBATCH -o ./RunOutput/yolov9_run.%j.out       # 作业标准输出文件
#SBATCH -e ./RunOutput/yolov9_run.%j.err       # 作业标准错误文件
#SBATCH --partition=debug
#SBATCH --time=0:05:00
#SBATCH --mail-type=ALL
#SBATCH --mail-user=xxxxxxxxxxx@qq.com


# 激活Anaconda环境

module add apps/anaconda3
module add apps/cuda/11.4.4
source activate yolov09

#清空环境变量（不报错的话，可删去）
unset LD_LIBRARY_PATH


# 运行PyTorch任务
python -u yolov9/yolov9-main/train_dual.py
#python -u yolov9/yolov9-main/detect.py

注意：yolov09环境需要配置在  .conda/envs/   目录下

2.在终端输入以下命令
 

sbatch yolov9.sh

3.曙光超算slurm作业使用的常用命令
参考：曙光超算slurm作业使用_wide_resnet50_racm-8234f177.pth-CSDN博客