vaeac_check_cuda: Function that checks for access to CUDA

In shapr: Prediction Explanation with Dependence-Aware Shapley Values

Function that checks for access to CUDA

Description

Function that checks for access to CUDA

Usage

vaeac_check_cuda(cuda)

Arguments

cuda

Logical (default is FALSE). If TRUE, then the vaeac model will be trained using cuda/GPU. If torch::cuda_is_available() is FALSE, the we fall back to use CPU. If FALSE, we use the CPU. Using a GPU for smaller tabular dataset often do not improve the efficiency. See vignette("installation", package = "torch") fo help to enable running on the GPU (only Linux and Windows).

Value

The function does not return anything.

Author(s)

Lars Henry Berge Olsen

shapr documentation built on April 4, 2025, 12:18 a.m.