Models: 【object detection API】How to specify gpu card index?

[L-CY]

I am following the Quick Start to train a pet detector. I have a server with four gtx1080Ti. My question is how to specify which gpu card (ig. one card or two card) I wanna use?

[L-CY]

solved

[SimsGautam]

@L-CY How did you solve this?

[L-CY]

@SimsGautam You can specify CUDA_VISIBLE_DEVICES when running ur python code.

[ghost]

@L-CY, I do the same but it still seems to run on CPU. It would be helpful if you could specify where exactly in the code did you specify CUDA_VISIBLE_DEVICES.

Thanks!

[gsssrao]

Enter the following in your terminal to switch to gpu 1:
export CUDA_VISIBLE_DEVICES=1

[wander1985]

This solution works for me on Windows.
https://stackoverflow.com/a/49735789/11552884