指定要在其上运行训练的GPU（TF对象检测API，模型Zoo）

Question

我正在具有多个GPU的设备上训练对象检测，并且想在gpu 1上运行训练（保持0和2空闲），并且在开始训练时看不到这样做的选项。我已经查看了train.py和model_main.py，也找不到在其中更改的行。有什么建议吗？

Answer 1

使用CUDA_VISIBLE_DEVICES环境变量。 您可以通过在命令行中添加以下内容来做到这一点：

CUDA_VISIBLE_DEVICES=1 python <your-python-script>

它将仅启用TF的GPU 1。