在并行用gpu运行python文件

在并行用gpu运行python文件

 

1. vim 写脚本模板

#!/bin/bash
module load anaconda/2020.11
source activate 环境名
export PYTHONUNBUFFERED=1
python xxx.py

 

2.提交作业：

sbatch --gpus=卡数 脚本名

 

3.查看作业情况：

squeue

 

4.结束作业：

scancel  作业号（作业号执行squeue即可查看到）

 

5.实时查看输出文件：

tail -f   文件名