Function cuda_device_count 

pub fn cuda_device_count() -> usize

Get the number of CUDA devices available

This function returns the number of CUDA devices available on the system. The result depends on the CUDA feature flag and system configuration.

Returns

Number of CUDA devices available:

• 0 if CUDA feature is disabled
• 0 if CUDA is not available on the system
• Number of available CUDA devices if CUDA is available

Thread Safety

This function is thread-safe and can be called from multiple threads.

Examples

use train_station::{cuda_device_count, Device};

let device_count = cuda_device_count();
println!("Found {} CUDA devices", device_count);

for i in 0..device_count {
    let device = Device::cuda(i);
    println!("CUDA device {}: {}", i, device);
}