cuda.h

#pragma once

#include <torch/csrc/WindowsTorchApiMacro.h>

#include <cstddef>

namespace torch {
namespace cuda {
size_t TORCH_API device_count();

bool TORCH_API is_available();

bool TORCH_API cudnn_is_available();
} // namespace cuda
} // namespace torch