mojolearn.gpu_arch

mojolearn.gpu_arch()

The GPU architecture whose binaries this process loaded.

Returns:

str or None – The architecture name, such as 'sm_80' or 'gfx942'; None when no architecture choice was made (macOS, a source checkout, or a CPU-only install).

Examples

>>> import mojolearn
>>> arch = mojolearn.gpu_arch()