Unverified Commit e8b126d9 authored by Stas Bekman's avatar Stas Bekman Committed by GitHub
Browse files

[build] add compute_86 (#577)

RTX-30 series are compute_86
```
python -c "import torch; print(torch.cuda.get_device_capability())"
```
This PR adds support for this compute capability.

Reference: https://developer.nvidia.com/cuda-gpus

Co-authored-by: default avatarJeff Rasley <jerasley@microsoft.com>
parent ce363d0e
......@@ -34,7 +34,7 @@ def installed_cuda_version():
def get_default_compute_capatabilities():
compute_caps = DEFAULT_COMPUTE_CAPABILITIES
if installed_cuda_version()[0] >= 11:
compute_caps += ";8.0"
compute_caps += ";8.0;8.6"
return compute_caps
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment