def __queryGPUProcessStats(self):
        """
        Query stats for all GPU processes.

        Returns:
            gpuProcessStats (list): GPU process statistics.
        """
        processes = GPUtil.getGPUProcesses()
        if processes:
            return [
                GPUProcessStat(proc.pid, proc.processName, proc.gpuId,
                               proc.gpuUuid, proc.usedMemory)
                for proc in processes
            ]
        return []
Esempio n. 2
0
# Print os and python version information
print('OS: ' + sys.platform)
print(sys.version)

# Print package name and version number
print(GPU.__name__ + ' ' + GPU.__version__)

# Get all GPUs
gpus = GPU.getGPUs()
print("All gpus:")
for gpu in gpus:
    print(gpu)
print()

# Get all GPU processes
gpuProcesses = GPU.getGPUProcesses()
print("All gpu processes:")
for gpuProcess in gpuProcesses:
    print(gpuProcess)
print()

# Show the utilization of all GPUs in a nice table
GPU.showUtilization()

# Show all stats of all GPUs in a nice table
GPU.showUtilization(all=True)

# Get all available GPU(s), ordered by ID in ascending order
print('All available ordered by id: '),
print(GPU.getAvailable(order='first', limit=999))