未验证 提交 e8b126d9 编写于 作者: S Stas Bekman 提交者: GitHub

[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-gpusCo-authored-by: NJeff Rasley <jerasley@microsoft.com>
上级 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% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册