前往新版Gitcode,体验更适合开发者的 AI 搜索 >>
Created by: yyhlvdl
hi,我最近在训练模型。我们实验室有4个gpu,我只能使用其中的3个,然而,我在train.py中设置train counter为3后,它自动选择编码为0,1,2的gpu,然而,我只能使用1,2,3的gpu,有什么方法可以设置吗?