cuda_cores_per_mp KeyError: 7.5

[JingchaoZhang]

I got the following error when running the python deviceQuery.py command.

CUDA device query (PyCUDA version)

Detected 1 CUDA Capable device(s)

Device 0: Quadro T1000 with Max-Q Design
         Compute Capability: 7.5
         Total Memory: 4096 megabytes
Traceback (most recent call last):
  File "deviceQuery.py", line 35, in <module>
    cuda_cores_per_mp = { 5.0 : 128, 5.1 : 128, 5.2 : 128, 6.0 : 64, 6.1 : 128, 6.2 : 128}[compute_capability]
KeyError: 7.5

Can you update the code to include key 7.5 in the dict?

[Harsh188]

I have a similar issue as well, however, my KeyError value is 8.6.

[YuhsiHu]

You can add the version and cores in to the dict cuda_cores_per_mp. For example, add:
7.5 : 128

[foongminwong]

Adding 8.6 : 128 to cuda_cores_per_mp works, thanks!

FYI, if using Python3, replace gpu_device.get_attributes().iteritems() to gpu_device.get_attributes().items()