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!