Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Set TORCH_CUDA_ARCH_LIST to 8.0 and above
Only compiling on Compute Capability 8.0 and above, see https://developer.nvidia.com/cuda-gpus. I.e. NVIDIA Ampere generation devices or newer.
- Loading branch information