mirror of
https://github.com/open-mmlab/mmengine.git
synced 2025-06-03 21:54:44 +08:00
19 lines
323 B
ReStructuredText
19 lines
323 B
ReStructuredText
|
.. role:: hidden
|
||
|
|
:class: hidden-section
|
||
|
|
|
||
|
|
mmengine.device
|
||
|
|
===================================
|
||
|
|
|
||
|
|
.. currentmodule:: mmengine.device
|
||
|
|
|
||
|
|
.. autosummary::
|
||
|
|
:toctree: generated
|
||
|
|
:nosignatures:
|
||
|
|
:template: classtemplate.rst
|
||
|
|
|
||
|
|
get_device
|
||
|
|
get_max_cuda_memory
|
||
|
|
is_cuda_available
|
||
|
|
is_mlu_available
|
||
|
|
is_mps_available
|