Environment

System platform

Linux

Windows

MacOS

Device

NVIDIA GPU

For NVIDIA GPUs, we recommend using CUDA>=11.8

Install torch with CUDA support and test with

import torch
assert torch.cuda.is_available() is True

Ampere Architecture

Volta Architecture

Ascend NPU

Use torch-npu

AMD GPU

We support torch rocm version

Apple M2

install torch and test mps with

import torch  
assert torch.backends.mps.is_available() is True