使用pycuda检查gpu，验证pycuda安装成功

最新推荐文章于 2024-08-03 10:24:48 发布

Mr_pooh

最新推荐文章于 2024-08-03 10:24:48 发布

阅读量588

点赞数

文章标签： python 深度学习开发语言

本文链接：https://blog.csdn.net/Mr_pooh/article/details/129468654

版权

import pycuda.driver as drv

drv.init()
print('Detected {} CUDA Capable device(s)'.format(drv.Device.count()))

for i in range(drv.Device.count()):
    gpu_device=drv.Device(i)
    print('Device{}:{}'.format(i,gpu_device.name()))
    compute_capability=float('%d.%d' % gpu_device.compute_capability())
    print('\t Compute Capability: {}'.format(compute_capability))
    print('\t Total Memory: {} megabytes'.format(gpu_device.total_memory()//(1024**2)))
device_attributes_tuples = gpu_device.get_attributes().items()
device_attributes={}
for k,v in   device_attributes_tuples:
        device_attributes[str(k)]=v
num_mp=device_attributes['MULTIPROCESSOR_COUNT']
cuda_cores_per_mp= { 5.0:128,5.1:128,5.2:128,6.0:64,6.1:128,6.2:128}.setdefault('compute_capability',0)
print('\t ({}) Mutiprocessors,({}) CUDA Cores / Multiprocessor:{} CUDA Cores'.format(
      num_mp,cuda_cores_per_mp,num_mp*cuda_cores_per_mp))
device_attributes.pop('MULTIPROCESSOR_COUNT')
for k in device_attributes.keys():
      print( '\t {}:{}'.format(k,device_attributes[k]))

运行上面这段python脚本就可以了