解决问题:ValueError: Unknown CUDA arch (8.9) or GPU not supported

换新电脑了还没有markdown,凑合看

平台是ubuntu20.04

在开始训练Deformable-detr时编译环境出现的问题

进入你创建的的conda环境文件,我的是如下,

/home/anti1hapi/anaconda3/envs/deformable_detr/lib/python3.7/site-packages/torch/utils

找到其中的cpp_extension.py文件,修改其中的内容

# Note: keep combined names ("arch1+arch2") above single names, otherwise
# string replacement may not do the right thing
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'),
])

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.9']

大概是1300行左右,因为缺少的是8.9,所以我在下面添加了Ada,8.9,还有下面的版本,修改后问题解决,

如果仍然不行,我建议看看别的博主的修改bashrc文件的教程,因为我之前修改过这个文件

posted @ 2025-02-22 11:51  anti1hapi  阅读(590)  评论(0)    收藏  举报