Tensor float 32

Creator
Creator
Seonglae Cho
Created
Created
2021 May 13 9:52
Editor
Edited
Edited
2024 Mar 8 8:11
Refs
Refs
Language
Language
torch.backends.cuda.matmul.allow_tf32 = True # allow tf32 on matmul torch.backends.cudnn.allow_tf32 = True # allow tf32 on cudnn
 
 
 
 
 
 

Recommendations