标签: tensorflow installation gpu
我正在具有多个GPU的设备上训练对象检测,并且想在gpu 1上运行训练(保持0和2空闲),并且在开始训练时看不到这样做的选项。我已经查看了train.py和model_main.py,也找不到在其中更改的行。有什么建议吗?
答案 0 :(得分:1)
使用CUDA_VISIBLE_DEVICES环境变量。 您可以通过在命令行中添加以下内容来做到这一点:
CUDA_VISIBLE_DEVICES=1 python <your-python-script>
它将仅启用TF的GPU 1。