package torch

  1. Overview
  2. Docs
val device_count : unit -> int
val is_available : unit -> bool
val cudnn_is_available : unit -> bool
val set_benchmark_cudnn : bool -> unit
OCaml

Innovation. Community. Security.