Skip to main content

How to selected a GPU

Select one free GPU

    On this example we choose one GPU with at least 8192 MB memory
    #!/bin/bash
    
    #SBATCH --partition=gpu
    #SBATCH --gres=gpu
    #SBATCH --mem=8192MB
    
    COMMON=/usr/local/cuda/samples/common
    SAMPLE=/usr/local/cuda/samples/5_Simulations/nbody
    
    [ -d ../common ] || cp -r $COMMON ..
    [ -d nbody     ] || cp -r $SAMPLE .
    
    module load cuda
    cd nbody
    make clean
    make
    
    if [ -e nbody ]; then
    	chmod u+x nbody
    	./nbody -benchmark -numbodies=2560000
    fi
    

    Select a specific GPU: V100

    #!/bin/bash
    
    #SBATCH --partition=gpu
    #SBATCH --gres=gpu:v100s
    #SBATCH --mem=8192MB
    
    COMMON=/usr/local/cuda/samples/common
    SAMPLE=/usr/local/cuda/samples/5_Simulations/nbody
    
    [ -d ../common ] || cp -r $COMMON ..
    [ -d nbody     ] || cp -r $SAMPLE .
    
    module load cuda
    cd nbody
    make clean
    make
    
    if [ -e nbody ]; then
    	chmod u+x nbody
    	./nbody -benchmark -numbodies=2560000
    fi
    

    GPU list

    You can find the full GPU list per cluster here