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))