Up – Package index » torch » Sources » torch » device.ml

Source file `device.ml`

torch
- README
- Library torch
  - Torch
- Library torch.core
  - Torch_core
- Library torch.toplevel
  - Torch_toplevel
- Library torch.vision
  - Torch_vision
- Sources

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

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