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: Jeff Rasley <jerasley@microsoft.com>
This commit is contained in:
Stas Bekman 2020-12-07 12:49:50 -08:00 коммит произвёл GitHub
Родитель ce363d0e06
Коммит e8b126d986
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
1 изменённых файлов: 1 добавлений и 1 удалений

Просмотреть файл

@ -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