Module cuda 

Functions

atc_cuda_device_count^⚠

Returns the number of CUDA devices available.

atc_cuda_is_available^⚠

Returns true if at least one CUDA device is available.

atc_cudnn_is_available^⚠

Returns true if CUDA is available, and CuDNN is available.

atc_manual_seed^⚠

Sets the seed for the current GPU.

atc_manual_seed_all^⚠

Sets the seed for all available GPUs.

atc_set_benchmark_cudnn^⚠

Sets CUDNN benchmark mode.

atc_set_user_enabled_cudnn^⚠

Enable or disable CUDNN.

atc_synchronize^⚠

Waits for all kernels in all streams on a CUDA device to complete.

atc_user_enabled_cudnn^⚠

Returns true if CUDNN is enabled by the user.