Needed tweaks to get a GPU to recognize the code

[epowell101]

The Bert4Eth model uses TensorFlow v 1 - and disables v 2. This may have something to do with why I had a hard time getting the code to recognize available GPUs.

Ultimately I added the following to help with troubleshooting and assigning the GPU:
gpus = tf.config.list_physical_devices('GPU')
if gpus:
try:
tf.config.set_visible_devices(gpus[0], 'GPU')
logical_gpus = tf.config.list_logical_devices('GPU')
print(len(logical_gpus), "Physical GPUs,", len(logical_gpus), "Logical GPU")
except RuntimeError as e:
print(e)

[epowell101]

Current environment that may be working:
name: w_numpy_119
channels:

• defaults
• conda-forge
  dependencies:
• python
• tensorflow-gpu=1.15
• numpy 1.19.5
• cudatoolkit
• cudnn
• pip