Think about how to organize the visible GPU devices using:
os.environ["CUDA_VISIBLE_DEVICES"] = "0, 1, ..."