pycuda-查看GPU的基本属性

import pycuda.driver as cuda
from six.moves import range

cuda.init()
print("%d device(s) found." % cuda.Device.count())

for ordinal in range(cuda.Device.count()):
    dev = cuda.Device(ordinal)
    print('Device #%d: %s' % (ordinal, dev.name()))
    print(' Compute Capability: %d.%d' % dev.compute_capability())
    print(' Total Memory: %s KB' % (dev.total_memory()//(1024)))

    atts = [ (str(att), value) for att, value in list(dev.get_attributes().items())]
    atts.sort()

    for att, value in atts:
        print(' %s : %s' % (att, value))

1 device(s) found.
Device #0: GeForce GTX 960M
 Compute Capability: 5.0
 Total Memory: 4194304 KB
 ASYNC_ENGINE_COUNT : 1
 CAN_MAP_HOST_MEMORY : 1
 CLOCK_RATE : 1176000
 COMPUTE_CAPABILITY_MAJOR : 5
 COMPUTE_CAPABILITY_MINOR : 0
 COMPUTE_MODE : DEFAULT
 CONCURRENT_KERNELS : 1
 ECC_ENABLED : 0
 GLOBAL_L1_CACHE_SUPPORTED : 0
 GLOBAL_MEMORY_BUS_WIDTH : 128
 GPU_OVERLAP : 1
 INTEGRATED : 0
 KERNEL_EXEC_TIMEOUT : 1
 L2_CACHE_SIZE : 2097152
 LOCAL_L1_CACHE_SUPPORTED : 1
 MANAGED_MEMORY : 1
 MAXIMUM_SURFACE1D_LAYERED_LAYERS : 2048
 MAXIMUM_SURFACE1D_LAYERED_WIDTH : 16384
 MAXIMUM_SURFACE1D_WIDTH : 16384
 MAXIMUM_SURFACE2D_HEIGHT : 65536
 MAXIMUM_SURFACE2D_LAYERED_HEIGHT : 16384
 MAXIMUM_SURFACE2D_LAYERED_LAYERS : 2048
 MAXIMUM_SURFACE2D_LAYERED_WIDTH : 16384
 MAXIMUM_SURFACE2D_WIDTH : 65536
 MAXIMUM_SURFACE3D_DEPTH : 4096
 MAXIMUM_SURFACE3D_HEIGHT : 4096
 MAXIMUM_SURFACE3D_WIDTH : 4096
 MAXIMUM_SURFACECUBEMAP_LAYERED_LAYERS : 2046
 MAXIMUM_SURFACECUBEMAP_LAYERED_WIDTH : 16384
 MAXIMUM_SURFACECUBEMAP_WIDTH : 16384
 MAXIMUM_TEXTURE1D_LAYERED_LAYERS : 2048
 MAXIMUM_TEXTURE1D_LAYERED_WIDTH : 16384
 MAXIMUM_TEXTURE1D_LINEAR_WIDTH : 134217728
 MAXIMUM_TEXTURE1D_MIPMAPPED_WIDTH : 16384
 MAXIMUM_TEXTURE1D_WIDTH : 65536
 MAXIMUM_TEXTURE2D_ARRAY_HEIGHT : 16384
 MAXIMUM_TEXTURE2D_ARRAY_NUMSLICES : 2048
 MAXIMUM_TEXTURE2D_ARRAY_WIDTH : 16384
 MAXIMUM_TEXTURE2D_GATHER_HEIGHT : 16384
 MAXIMUM_TEXTURE2D_GATHER_WIDTH : 16384
 MAXIMUM_TEXTURE2D_HEIGHT : 65536
 MAXIMUM_TEXTURE2D_LINEAR_HEIGHT : 65536
 MAXIMUM_TEXTURE2D_LINEAR_PITCH : 1048544
 MAXIMUM_TEXTURE2D_LINEAR_WIDTH : 65536
 MAXIMUM_TEXTURE2D_MIPMAPPED_HEIGHT : 16384
 MAXIMUM_TEXTURE2D_MIPMAPPED_WIDTH : 16384
 MAXIMUM_TEXTURE2D_WIDTH : 65536
 MAXIMUM_TEXTURE3D_DEPTH : 4096
 MAXIMUM_TEXTURE3D_DEPTH_ALTERNATE : 16384
 MAXIMUM_TEXTURE3D_HEIGHT : 4096
 MAXIMUM_TEXTURE3D_HEIGHT_ALTERNATE : 2048
 MAXIMUM_TEXTURE3D_WIDTH : 4096
 MAXIMUM_TEXTURE3D_WIDTH_ALTERNATE : 2048
 MAXIMUM_TEXTURECUBEMAP_LAYERED_LAYERS : 2046
 MAXIMUM_TEXTURECUBEMAP_LAYERED_WIDTH : 16384
 MAXIMUM_TEXTURECUBEMAP_WIDTH : 16384
 MAX_BLOCK_DIM_X : 1024
 MAX_BLOCK_DIM_Y : 1024
 MAX_BLOCK_DIM_Z : 64
 MAX_GRID_DIM_X : 2147483647
 MAX_GRID_DIM_Y : 65535
 MAX_GRID_DIM_Z : 65535
 MAX_PITCH : 2147483647
 MAX_REGISTERS_PER_BLOCK : 65536
 MAX_REGISTERS_PER_MULTIPROCESSOR : 65536
 MAX_SHARED_MEMORY_PER_BLOCK : 49152
 MAX_SHARED_MEMORY_PER_MULTIPROCESSOR : 65536
 MAX_THREADS_PER_BLOCK : 1024
 MAX_THREADS_PER_MULTIPROCESSOR : 2048
 MEMORY_CLOCK_RATE : 2505000
 MULTIPROCESSOR_COUNT : 5
 MULTI_GPU_BOARD : 0
 MULTI_GPU_BOARD_GROUP_ID : 0
 PCI_BUS_ID : 2
 PCI_DEVICE_ID : 0
 PCI_DOMAIN_ID : 0
 STREAM_PRIORITIES_SUPPORTED : 1
 SURFACE_ALIGNMENT : 512
 TCC_DRIVER : 0
 TEXTURE_ALIGNMENT : 512
 TEXTURE_PITCH_ALIGNMENT : 32
 TOTAL_CONSTANT_MEMORY : 65536
 UNIFIED_ADDRESSING : 1
 WARP_SIZE : 32