[RLlib] GPU selection

Hi again, I’m working with RLlib in a system that has two available Nvidia GPUs. Is there any way to choose with one to use for model training? I mean, if I want to compare individual performance with both GPUs I want to train using one of them fist and the other one later and compare times in both cases. The only way that I found was to change tensorflow visible devices with tf.config.set_visible_devices(), but I wanted to know if this is the correct way to do that.

Thanks in advance!

@sven1977 is this the correct way to do it?