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.