def get_device_count(self): """ Return count of Nvidia devices """ if is_macos: self.device_count = pynvx.cudaDeviceGetCount(ignore=True) else: try: self.device_count = pynvml.nvmlDeviceGetCount() except pynvml.NVMLError: self.device_count = 0
def get_device_count(self): """ Return count of Nvidia devices """ if IS_MACOS: self.device_count = pynvx.cudaDeviceGetCount(ignore=True) else: try: self.device_count = pynvml.nvmlDeviceGetCount() except pynvml.NVMLError: self.device_count = 0 if self.logger: self.logger.debug("GPU Device count: %s", self.device_count)
def get_device_count(self): """ Return count of Nvidia devices """ if IS_MACOS: self.device_count = pynvx.cudaDeviceGetCount(ignore=True) else: try: self.device_count = pynvml.nvmlDeviceGetCount() except pynvml.NVMLError: self.device_count = 0 if self.logger: self.logger.debug("GPU Device count: %s", self.device_count)
def _get_device_count(self) -> int: """ Detect the number of GPUs attached to the system. Returns ------- int The total number of GPUs connected to the PC """ retval = pynvx.cudaDeviceGetCount(ignore=True) # pylint:disable=no-member self._log("debug", f"GPU Device count: {retval}") return retval
def _get_device_count(self): """ Detect the number of GPUs attached to the system and allocate to :attr:`_device_count`. """ if self._is_plaidml: self._device_count = self._plaid.device_count elif IS_MACOS: self._device_count = pynvx.cudaDeviceGetCount(ignore=True) else: try: self._device_count = pynvml.nvmlDeviceGetCount() except pynvml.NVMLError: self._device_count = 0 self._log("debug", "GPU Device count: {}".format(self._device_count))
def print_info(): try: pynvx.cudaInit() except RuntimeError as e: print(e) return print('================ CUDA INFO =====================') print('Driver Version : {}'.format(pynvx.cudaSystemGetDriverVersion())) print('Runtime Version : {}'.format(pynvx.cudaSystemGetRuntimeVersion())) print('Device Count : {}'.format(pynvx.cudaDeviceGetCount())) handles = pynvx.cudaDeviceGetHandles() for handle in handles: print('------------------------------------------------') print('Device {}:'.format(handle)) print('Device Name : {}'.format( pynvx.cudaGetName(handle))) print('Device ClockRate : {} MHz'.format( pynvx.cudaGetClockRate(handle) / 1024)) print('Device ComputeCapability : {}'.format( pynvx.cudaGetComputeCapability(handle))) print('Device ProcessorCount : {}'.format( pynvx.cudaGetMultiProcessorCount(handle))) print('Device PciBusID : {}'.format( pynvx.cudaGetPciBusID(handle))) print('Device PciDeviceID : {}'.format( pynvx.cudaGetPciDeviceID(handle))) print('Device PciDomainID : {}'.format( pynvx.cudaGetPciDomainID(handle))) print('Device MemTotal : {} MiB'.format( pynvx.cudaGetMemTotal(handle) / (1024 * 1024))) print('Device MemFree : {} MiB'.format( pynvx.cudaGetMemFree(handle) / (1024 * 1024))) print('Device MemUsed : {} MiB'.format( pynvx.cudaGetMemUsed(handle) / (1024 * 1024)))
def test_device_count(): v = m.cudaDeviceGetCount(ignore=True) assert isinstance(v, int)