Skip to content
  • Code Reference
  • careamics_napari
  • utils
  • gpu_utils

gpu_utils

Utilities to test GPU availability with torch.

is_gpu_available() #

Check if GPU is available with torch.

Returns:

Type Description
bool

True if GPU is available, False otherwise.
Source code in src/careamics_napari/utils/gpu_utils.py 
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
def is_gpu_available() -> bool:
    """Check if GPU is available with torch.

    Returns
    -------
    bool
        True if GPU is available, False otherwise.
    """
    if platform.system() == "Darwin":
        # adapted from Lightning
        # pytorch-lightning/src/lightning/fabric/accelerators/mps.py
        mps_disabled = os.getenv("DISABLE_MPS", "0") == "1"
        return (
            not mps_disabled
            and backends.mps.is_available()
            and platform.processor() in ("arm", "arm64")
        )
    else:
        return cuda.is_available()