Print the TORCH_CUDA_ARCH_LIST

NVIDIA / apex

A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch

BSD 3-Clause "New" or "Revised" License

8.2k stars 1.36k forks source link

Open yncxcw opened 11 months ago

yncxcw commented 11 months ago

yncxcw commented 11 months ago

can someone help to review ?