1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
| #pragma once
|
| #include <torch/csrc/WindowsTorchApiMacro.h>
|
| #include <cstddef>
|
| namespace torch {
| namespace cuda {
| /// Returns the number of CUDA devices available.
| size_t TORCH_API device_count();
|
| /// Returns true if at least one CUDA device is available.
| bool TORCH_API is_available();
|
| /// Returns true if CUDA is available, and CuDNN is available.
| bool TORCH_API cudnn_is_available();
| } // namespace cuda
| } // namespace torch
|
|
