Choose single GPU for training

[AnhMinhTran]

As per the title, does anyone know how to choose only 1 GPU for training? E.g I choose device:1 out of (0,1)

[rwightman]

lookup CUDA_VISIBLE_DEVICES it's the recommended way to control device visibility at per-process granularity