Using PSC Bridge

$ module load gcc/7.2.0
$ interact -p GPU --gres=gpu:p100:2 -t 0:30:00 -N 1

(base) [qiangz@login005 CSP_BO]$ cat myrun_psc 
#!/bin/bash
#SBATCH -N 1
#SBATCH -p GPU
#SBATCH --ntasks-per-node 28
#SBATCH -t 0:30:00
#SBATCH --gres=gpu:p100:2

# echo commands to stdout
set -x

# move to working directory
# this job assumes:
# - all input data is stored in this directory
# - all output should be stored in this directory
#cd /pylon5/dmr200039p/qiangz

# run GPU program
python example_validate.py models/test_2.json database/PtHO.db > log

Sample interact command for GPU-AI

To run in an interactive session on Bridges-AI, use the interact command and specify the GPU-AI partition. An example interact command to request 1 GPU on a Volta 16 node is:

interact -p GPU-AI --gres=gpu:volta16:1

Where:

• -p indicates the intended partition

• --gres=gpu:volta16:1 requests the use of 1 V100 GPU on an Apollo node