schrodinger.infra.gpgpu_utils module¶
-
schrodinger.infra.gpgpu_utils.
get_local_gpgpus
()¶ Detect local GPUs via schrodinger.gpgpu
- Returns
list of tuple of local gpgpu device number and description
- Return type
list(tuple(dev_num, desc))
-
schrodinger.infra.gpgpu_utils.
get_remote_gpgpus
(host)¶ Return a list of GPGPUs belonging to the specified host.
- Parameters
host (str) – name of host matching a name from the host file
- Returns
list of tuple of gpgpu device number and description
- Return type
list(tuple(dev_num, desc))