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 found, return True else return False.

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