MultiScaleDeformableAttention编译报错解决

cuda版本是10.2  安装对应的pytorch

pip install torch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2
然后注释掉ops文件夹下setup.py中的

        #'-gencode', 'arch=compute_80,code=compute_80',
        #'-gencode', 'arch=compute_86,code=compute_86',
这两行代码  我是Nvidia RTX TITAN 显卡不一样这一步也不一样

编译成功

你可能感兴趣的:(深度学习)