Shortcuts

mmengine.device.get_device

mmengine.device.get_device()[source]

Returns the currently existing device type.

Returns

cuda | npu | mlu | mps | cpu.

Return type

str

Read the Docs v: latest
Versions
latest
stable
Downloads
pdf
html
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.