package torch

  1. Overview
  2. Docs

Module C.CudaSource

Sourceval device_count : (unit -> int Torch_generated.return) Torch_generated.result
Sourceval is_available : (unit -> int Torch_generated.return) Torch_generated.result
Sourceval cudnn_is_available : (unit -> int Torch_generated.return) Torch_generated.result
Sourceval set_benchmark_cudnn : (int -> unit Torch_generated.return) Torch_generated.result