schrodinger.infra.gpgpu_utils module¶

schrodinger.infra.gpgpu_utils.get_local_gpgpus()¶

Detect local GPUs via schrodinger.gpgpu

schrodinger.infra.gpgpu_utils.check_for_unsupported_graphics_card()¶

Check whether there are any graphics card with an unsupported compute capability (usually means they are too new).

If one is found, return the name of the card, otherwise return None.

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