How to configure the number of GPUs being used?

JDAI-CV / image-captioning

Implementation of 'X-Linear Attention Networks for Image Captioning' [CVPR 2020]

268 stars 52 forks source link

Closed nathanielhobbs closed 2 years ago

nathanielhobbs commented 2 years ago

I have a system with 10 GPUs, and when I start training, multiple (4) GPUs are used for processing.

Where can I configure the number of GPUs being used? e.g. if I only have 2 free GPUs of the 10
How can I specify which GPUs are being used? e.g. GPUs 0-5 are occupied, but I'd like to train on GPUs 6 and above.

nathanielhobbs commented 2 years ago

A1. at the command line set e.g. export CUDA_VISIBLE_DEVICES=0,3

A2. CUDA_VISIBLE_DEVICES has been hardcoded into the train.sh files.