enable_cudnn

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

Enable backend to use the cuDNN library.

Parameters:
  • enabled (bool, optional, default=True) – Use cuDNN library or not.
  • benchmark (bool, optional, default=False) – Select algorithms according to the benchmark or not.