fastdev.utils.cuda¶

Module Contents¶

fastdev.utils.cuda.cuda_toolkit_available() → bool[source]¶

Check if the nvcc is avaiable on the machine.

Return type:: bool

fastdev.utils.cuda.current_cuda_arch() → str[source]¶

Get the current CUDA architecture.

Return type:: str