Установка ограничения в сценарии задания 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 или выше?

Я следовал этому руководству по настройке ограничений.

0 ответов

Другие вопросы по тегам