Function that checks for access to CUDA
vaeac_check_cuda(cuda)
The function does not return anything.
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).
Lars Henry Berge Olsen