my desktop use two gpu. occurred error that 'Memory growth cannot differ between GPU devices'

[Yeong-Seo]

When i ran python train.py --mode="fit" --cfg_path="./configs/arc-res50.yaml", this error is occurred.

i found this code in train.py

so i thnk that two gpu have to set_memoy_growth() but 'flags.DEFINE_string('gpu', '0', 'which gpu to use')' is only one gpu. how can i do?

[mahendran-narayanan]

Use os.environ["CUDA_VISIBLE_DEVICES"]= "0" #gpu-id