若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 |