

torch.cuda.current_device


torch.cuda.current_stream


torch.cuda.default_stream


device


torch.cuda.device_count


device_of


torch.cuda.get_arch_list


torch.cuda.get_device_capability


torch.cuda.get_device_name


torch.cuda.get_device_properties


torch.cuda.get_gencode_flags


torch.cuda.get_sync_debug_mode


torch.cuda.init


torch.cuda.ipc_collect


torch.cuda.is_available


torch.cuda.is_initialized


torch.cuda.memory_usage


torch.cuda.set_device


torch.cuda.set_stream


torch.cuda.set_sync_debug_mode


torch.cuda.stream


torch.cuda.synchronize


torch.cuda.utilization


torch.cuda.OutOfMemoryError


Random Number Generator


Communication collectives


Streams and events


Graphs (beta)


Memory management


NVIDIA Tools Extension (NVTX)


Jiterator (beta)


Stream Sanitizer (prototype)


torch.backends


torch.backends.cuda


torch.backends.cudnn


torch.backends.mps

