A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
BSD 3-Clause "New" or "Revised" License
8.16k
stars
1.35k
forks
source link
Only print the warning message about `TORCH_CUDA_ARCH_LIST` if not set #1812
Open
aurianer opened 1 week ago