怎么控制GPU的序号

[zkh123456789]

作者您好，我想知道训练的时候怎么指定GPU的型号，每次我使用指令CUDA_VISIBLE_DEVICES=2,3这条指令的时候，似乎不管用，还是使用的默认的前两块GPU，请问您如何解决这个问题。

[CV-JunchengLi]

在trainer.py加

import os

os.environ["CUDA_VISIBLE_DEVICES"] = "4,5"