Skip to content

Instantly share code, notes, and snippets.

@berceanu

berceanu/numba_uuid.py

Last active February 8, 2021 17:47
Show Gist options
  • Save berceanu/8c70962a8c981ddf9e259135fbab294d to your computer and use it in GitHub Desktop.
Save berceanu/8c70962a8c981ddf9e259135fbab294d to your computer and use it in GitHub Desktop.
Download ZIP
Example usage of numba for getting GPU UUID
Raw
numba_uuid.py
from numba import cuda
with cuda.gpus[0]:
dev = cuda.get_current_device()
print(dev.uuid)
with cuda.gpus[1]:
dev = cuda.get_current_device()
print(dev.uuid)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment