Expand description
Device selection for CPU/CUDA
Functionsยง
- cuda_
is_ available - Check if CUDA is available
- get_
device - Get best available device (CUDA if available, otherwise CPU)
- get_
device_ info - Get device info string
- get_
device_ info_ string - get device information
- get_
device_ with_ preference - Get device with preference (for testing/debugging)
- log_
device_ info - print_
device_ info