schrodinger.gpgpu module

schrodinger::gpgpu C++ wrappers

class schrodinger.gpgpu.CudaDevice

schrodinger::gpgpu::CudaDevice

__init__(*args, **kwargs)

Initialize self. See help(type(self)) for accurate signature.

append()

appends another ‘this’ object

computeCapability() → std::array< int,2 >
computeMode() → std::string
cores() → int
description() → std::string
name() → std::string
next()

returns the next ‘this’ object

number() → int
uuid() → std::string
schrodinger.gpgpu.get_available_devices() → std::vector< schrodinger::gpgpu::CudaDevice,std::allocator< schrodinger::gpgpu::CudaDevice > > const &
schrodinger.gpgpu.get_minimum_nvml_driver() → std::string
schrodinger.gpgpu.get_scaled_token_count(host, default_tokens) → int
schrodinger.gpgpu.in_dev_env() → bool[source]

Return True if the process is running in a development environment - where SCHRODINGER_SRC or SCHRODINGER_DEV_DEBUG is set.

schrodinger.gpgpu.is_any_gpu_available(minimum_compute) → bool

is_any_gpu_available() -> bool

schrodinger.gpgpu.is_nvml_driver_supported() → bool
schrodinger.gpgpu.print_gpgpu_devices(verbose=True)

print_gpgpu_devices()

schrodinger.gpgpu.verify_any_gpu_available(minimum_compute)

verify_any_gpu_available()

schrodinger.gpgpu.verify_cuda_runtime()[source]

Check that we can run cuda programs.

Raise

RuntimeError if we can not find a usable GPU.

schrodinger.gpgpu.weakref_proxy()

proxy(object[, callback]) – create a proxy object that weakly references ‘object’. ‘callback’, if given, is called with a reference to the proxy when ‘object’ is about to be finalized.