schrodinger.gpgpu module¶
schrodinger::gpgpu C++ wrappers
- class schrodinger.gpgpu.CudaDevice¶
schrodinger::gpgpu::CudaDevice
- __init__(*args, **kwargs)¶
- computeCapability(self) std::array< int,2 >¶
- computeMode(self) std::string¶
- cores(self) int¶
- description(self) std::string¶
- discountedCores(self) int¶
- get_device_info(self) DeviceInfo¶
- mig_enabled(self) bool¶
- name(self) std::string¶
- number(self) int¶
- uuid(self) std::string¶
- class schrodinger.gpgpu.DeviceInfo¶
schrodinger::gpgpu::DeviceInfo
- __init__(*args, **kwargs)¶
- capability¶
- compute_mode¶
- cores¶
- name¶
- number¶
- pci_device_id¶
- uuid¶
- class schrodinger.gpgpu.MIGInfo¶
schrodinger::gpgpu::MIGInfo
- __init__(*args, **kwargs)¶
- cpu_count¶
- gpu_count¶
- memory_size_MB¶
- number¶
- sm_count¶
- uuid¶
- schrodinger.gpgpu.get_available_devices() std::vector< schrodinger::gpgpu::CudaDevice,std::allocator< schrodinger::gpgpu::CudaDevice > > const &¶
- schrodinger.gpgpu.get_device_uuid(pci_bus_id) std::string¶
- schrodinger.gpgpu.get_minimum_nvml_driver() std::string¶
- schrodinger.gpgpu.get_scaled_token_count(host, default_tokens) int¶
- schrodinger.gpgpu.in_dev_env() bool¶
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() bool¶
- schrodinger.gpgpu.print_gpgpu_devices(verbose=True)¶
- schrodinger.gpgpu.verify_any_gpu_available()¶
- schrodinger.gpgpu.verify_cuda_runtime()¶
- schrodinger.gpgpu.verify_nvml_driver_supported()¶