enable_cudnn

dragon.cuda.enable_cudnn(enabled=True)[source]

Activate the CuDNN engine.

Parameters:
  • enabled (bool, optional, default=True) – True to activate CuDNN.