How to find out if you have colab pro or not inside colab?

Question

. 。 Answers to this question are eligible for a +50<\/span> reputation bounty. . 寻找。

I just simply can't find any documentation on this topic. For instance, my usage would be like this:

"

Answer 1

#A Colab pro environment should have >20Gb of total memory.
from psutil import virtual_memory
colab_pro = virtual_memory().total / 1e9
print('Your runtime has {:.1f} gigabytes of available RAM\n'.format(colab_pro))

if colab_pro < 20:
  print('Not using a high-RAM runtime')
  # train model with lower settings
else:
  print('You are using a high-RAM runtime!')
  # train model with higher settings