:ambulance: Fix memory growth amongst multiple GPU

[DaemonOnCode]

Description

• While dealing with multiple GPUs, Memory Growth needs to be constant tf.config.set_visible_devices(physical_devices[0:1], 'GPU')

Reference

https://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/framework/config.py

Type of change

Please delete options that are not relevant.

• [x] Bug fix (non-breaking change which fixes an issue)
• [ ] New feature (non-breaking change which adds functionality)
• [x] Breaking change (fix or feature that would cause existing functionality to not work as expected)
• [x] This change requires a documentation update

Checklist:

• [x] My code follows the style guidelines of this project
• [x] I have performed a self-review of my own code
• [x] I have commented on my code, particularly in hard-to-understand areas
• [ ] I have made corresponding changes to the documentation
• [ ] My changes generate no new warnings
• [ ] I have added tests that prove my fix is effective or that my feature works
• [x] New and existing unit tests pass locally with my changes
• [ ] Any dependent changes have been merged and published in downstream modules