aicube的n卡gpu索引该如何添加

请问有人知道aicube怎样才能读取n卡的gpu索引呢，我已经安装了cuda和cudnn，在全局的py里添加了torch，能够调用gpu，当还是只能看到默认的gpu0，显示不了gpu1，gpu0是集显，训练速度太感人了