torch.backends

API名称

是否支持

限制与说明

torch.backends.cuda.is_built

  

torch.backends.cuda.matmul.allow_tf32

  

torch.backends.cuda.matmul.allow_fp16_reduced_precision_reduction

  

torch.backends.cuda.clear

  

torch.backends.cuda.preferred_linalg_library

  

torch.backends.cuda.SDPBackend

  

torch.backends.cuda.flash_sdp_enabled

  

torch.backends.cuda.enable_mem_efficient_sdp

  

torch.backends.cuda.mem_efficient_sdp_enabled

  

torch.backends.cuda.enable_flash_sdp

  

torch.backends.cuda.math_sdp_enabled

  

torch.backends.cuda.enable_math_sdp

  

torch.backends.cuda.sdp_kernel

  

torch.backends.cudnn.version

  

torch.backends.cudnn.is_available

  

torch.backends.cudnn.enabled

  

torch.backends.cudnn.allow_tf32

  

torch.backends.cudnn.deterministic

  

torch.backends.cudnn.benchmark

  

torch.backends.mps.is_available

  

torch.backends.mps.is_built

  

torch.backends.mkl.is_available

  

torch.backends.mkl.verbose

  

torch.backends.mkldnn.is_available

  

torch.backends.mkldnn.verbose

  

torch.backends.openmp.is_available

  

torch.backends.opt_einsum.is_available

  

torch.backends.opt_einsum.get_opt_einsum