Rate this Page
★ ★ ★ ★ ★

torch.cuda.is_tf32_supported#

torch.cuda.is_tf32_supported()[source]#

Return a bool indicating if the current CUDA/ROCm device supports dtype tf32.

Return type:

bool

On this page
Show Source
PyTorch Libraries