Select a GPU#

Today nabu works on a single GPU. If you have different processes requiring GPU it can be convenient to select a specific one and avoid conflicts.

You can do this either from a terminal before launching tomwer or from a ‘python script’ widget. On this example we have more n GPU (n>=0) and we want to use the first one.

From a python script widget#

import os
os.environ[CUDA_VISIBLE_DEVICES] = 0

From a terminal#

# define the GPU to be used
export CUDA_VISIBLE_DEVICES=0
# start tomwer canvas
tomwer canvas

Hint

You can list accessible GPU devices on a computer / slurm node by using nvidia-smi from a terminal

../../_images/nvidia_smi.png