1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
|
from __future__ import print_function
from __future__ import absolute_import
import pycuda.driver as drv
from six.moves import range
drv.init()
print("%d device(s) found." % drv.Device.count())
for ordinal in range(drv.Device.count()):
dev = drv.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))
|
