报错ValueError: Unknown CUDA arch (8.6) or GPU not supported

问题描述

报错 ValueError: Unknown CUDA arch (8.6) or GPU not supported

本人显卡为 RTX 3060，CUDA 为 10.2，PyTorch 为 1.5

解决方案

修改 C:\Users\Administrator\Envs\test\Lib\site-packages\torch\utils\cpp_extension.py 的 named_arches 和 supported_arches 尝试解决：

    named_arches = collections.OrderedDict([
        ('Kepler+Tesla', '3.7'),
        ('Kepler', '3.5+PTX'),
        ('Maxwell+Tegra', '5.3'),
        ('Maxwell', '5.0;5.2+PTX'),
        ('Pascal', '6.0;6.1+PTX'),
        ('Volta', '7.0+PTX'),
        ('Turing', '7.5+PTX'),
        ('Ampere', '8.0;8.6+PTX'),
        ('Ada', '8.9+PTX'),
        ('Hopper', '9.0+PTX'),
    ])

    supported_arches = ['3.5', '3.7', '5.0', '5.2', '5.3', '6.0', '6.1', '6.2',
                        '7.0', '7.2', '7.5', '8.0', '8.6', '8.7', '8.9', '9.0']

很有可能不能解决问题，接着编译报错 nvcc fatal : Unsupported gpu architecture 'compute_86'

建议安装该显卡支持的 CUDA 版本。

如 RTX 3060 的 Compute Capability 为 8.6，支持的 CUDA SDK 为 11.x

参考文献

1. 3080算力添加、ValueError: Unknown CUDA arch (8.6) or GPU not supported 解决RTX30 运行深度学习代码报错
2. pytorch - ValueError: Unknown CUDA arch (8.6) or GPU not supported
3. ValueError: Unknown CUDA arch (8.6) or GPU not supported 解决RTX30 运行深度学习代码报错
4. CUDA - Wikipedia
5. CUDA GPUs - Compute Capability