Установка ограничения в сценарии задания slurm для вычислительных возможностей графического процессора
Я пытаюсь установить ограничение, чтобы моя работа выполнялась только на графических процессорах с вычислительной мощностью выше (или равной) 7.
Вот мой сценарий с именем
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
но в любом случае я получаю следующую ошибку:
[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 или выше?
Я следовал этому руководству по настройке ограничений.