Python v0.3.0

cudnn

Properties

allow_tf32

dragon.vm.torch.backends.cudnn.allow_tf32 = False

The flag that allows cuDNN TF32 math type or not.

benchmark

dragon.vm.torch.backends.cudnn.benchmark = False

The flag that benchmarks fastest cuDNN algorithms or not.

deterministic

dragon.vm.torch.backends.cudnn.deterministic = False

The flag that selects deterministic cuDNN algorithms or not.

enabled

dragon.vm.torch.backends.cudnn.enabled = True

The flag that uses cuDNN or not.

Functions

is_available

cudnn.is_available()

Return a bool reporting if cuDNN is available.

Returns:
bool – True if available otherwise False.

version

cudnn.version()

Return the cuDNN version.

Returns:
int – The version number.