vaeac_check_cuda: Function that checks for access to CUDA

In NorskRegnesentral/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

NorskRegnesentral/shapr documentation built on April 19, 2024, 1:19 p.m.