torch.backends

若API未标明支持情况,则代表该API的支持情况待验证。

API名称

是否支持

限制与说明

torch.backends.cuda.is_built

     

torch.backends.cuda.matmul.allow_tf32

     

torch.backends.cuda.matmul.allow_fp16_reduced_precision_reduction

     

torch.backends.cuda.cufft_plan_cache.size

     

torch.backends.cuda.cufft_plan_cache.max_size

     

torch.backends.cuda.cufft_plan_cache.clear

     

torch.backends.cuda.preferred_linalg_library

     

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.mkl.is_available

     

torch.backends.mkldnn.is_available

     

torch.backends.openmp.is_available