nvidia-smi中gpu的id和指定CUDA_VISIBLE_DEVICES时的id是不同的
如下方式可以获取指定CUDA_VISIBLE_DEVICES时的实际id
from tensorflow.python.client import device_lib
local_device_protos = device_lib.list_local_devices()
参考资料:
https://stackoverflow.com/questions/38559755/how-to-get-current-available-gpus-in-tensorflow