schrodinger.infra.gpgpu_utils module

schrodinger.infra.gpgpu_utils.get_local_gpgpus()

Detect local GPUs via mmgpgpu

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