package torch

  1. Overview
  2. Docs
package torch
Legend:
Page
Library
Module
Module type
Parameter
Class
Class type
Source

Source file device.ml

1
2
3
4
5
6
7
8
9
type t = Torch_core.Device.t =
  | Cpu
  | Cuda of int

let cuda_if_available () = if Cuda.is_available () then Cuda 0 else Cpu

let is_cuda = function
  | Cpu -> false
  | Cuda _ -> true