| #SBATCH --job-name=sam2 | |
| #SBATCH --nodes=1 # Request 1 node | |
| #SBATCH --ntasks-per-node=1 | |
| #SBATCH --gpus-per-task=a100-40g:1 | |
| #SBATCH --cpus-per-gpu=8 # Number of CPU cores (threads) per task | |
| #SBATCH --mem-per-gpu=40G # Memory limit per CPU core (there is no --mem-per-task) | |
| #SBATCH --time=24:00:00 # Job timeout | |
| #SBATCH --output=sam2_result.log # Redirect stdout to a log file | |
| #SBATCH --nodelist=gcp-us-0 | |
| cd /home/yuqian_fu/Projects/sam2 | |
| srun --nodes "$SLURM_NNODES" --ntasks-per-node 1 -- \ | |
| mkenv -f sam2.yml -- \ | |
| sh -c " | |
| python vos_inference.py --output_mask_dir /data/work-gcp-europe-west4-a/yuqian_fu/Ego/sam_result | |
| " | |