Son correctas estas líneas para usar la gpu?

# If you need nodes with special features you can select a constraint.
# Please, use cal by default. You will be assigned a node that satisfies your requests.
##SBATCH --constraint=cal
 
# Change "cal" by "sd" if you want to use Intel nodes and by "sr" if you want to use AMD nodes.
##SBATCH --constraint=exa
#SBATCH --constraint=dgx

# To use GPU, comment out the constraint line and uncomment the following line.
#SBATCH --gres=gpu:1