package torch

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

Innovation. Community. Security.