module load xxx
来加载一些软件的环境包.bash#!/bin/bash
#SBATCH -o job.%j.out
#SBATCH -p gpu
#SBATCH -J train_test
#SBATCH --nodes=1
#SBATCH --ntasks-per-node=1
#SBATCH -G 1
#module load miniconda
#module load cuda/12.2
nvidia-smi
解释:
bash-p 即计算节点的分区 -o 计算作业的输出重定向到哪个文件 --nodes 一共需要几个节点 --ntasks-per-node 一个节点几个人物 --G 需要几张显卡
bashsrun --gres=gpu:1 --pty xxx