gpu - 在 slurm 作业脚本中设置 GPU 计算能力的约束

Question

我正在尝试设置一个约束，以便我的工作只能在计算能力高于（或等于）7 的 GPU 上运行。

这是我的脚本名为torch_gpu_sanity_venv385-11.slurm：

#!/bin/bash
#SBATCH --partition=gpu-L --gres=gpu:1 --constraint="cc7.0" 
# -------------------------> ask for 1 GPU
d=$(date)
h=$(hostname)
echo $d $h env         # show CUDA related Env vars 
env|grep -i cuda
# nvidia-smi
#          actual work
/research/jalal/slurm/fashion/fashion_compatibility/torch_gpu_sanity_venv385-11.bash 

不使用--constraint="cc7.0"我的脚本运行正确。我什至使用了另一个版本，--constraint=cc7.0但无论哪种情况，我都会收到以下错误：

[jalal@goku fashion_compatibility]$ sbatch torch_gpu_sanity_venv385-11.slurm 
sbatch: error: Batch job submission failed: Invalid feature specification

当我删除该--constraint="cc7.0"术语时，我可以运行该作业。删除约束项后：

[jalal@goku fashion_compatibility]$ sbatch torch_gpu_sanity_venv385-11.slurm 
Submitted batch job 28398

那么，如何设置约束，以便只分配具有 7 或更高计算能力的 GPU？

我按照本教程进行约束设置。