#!/bin/bash
#SBATCH --partition=dgxa100
#SBATCH --gres=gpu:1
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
nvcc -o hello_cuda hello_cuda.cu
./hello_cuda