Check PyTorch device availability — check

This function prints the available devices (cpu, cuda, mps) detected by PyTorch. If your mps/cuda device is not shown, check your PyTorch installation.

Usage

check_devices(env_path = NULL)

Arguments

env_path: Path to virtual environment containing PyTorch and ciss-vae. Defaults to NULL.

Value

Vector of strings for available devices.

Examples

# \donttest{
try(
check_devices()
)# }
#> Error in py_module_import(module, convert = convert) : 
#>   ImportError: DLL load failed while importing _C: The specified module could not be found.
#> Run `reticulate::py_last_error()` for details.