3090编译Ocean可变形卷积 / Compile Ocean DCN on 3090 深度学习pytorch算法

see https://github.com/Soarkey/tr...

1.环境 / Environment

Ubuntu 20
Python 3.7
CUDA 11.4
PyTorch 1.9.0+cu111

2.修改 / Modification 找到lib/models/dcn/src/deform_conv.py文件, 将所有的AT_CHECK替换为TORCH_CHECK, 同时将所有的.view操作替换为.reshape
操作, 之后重新执行编译python setup.py develop.
【3090编译Ocean可变形卷积 / Compile Ocean DCN on 3090】Found file lib/models/dcn/src/deform_conv.py, replace all AT_CHECK and .view operations with TORCH_CHECK and .reshape, then recompile python setup.py develop.
3.可能碰到的问题及解决 / References > ValueError: Unknown CUDA arch (8.6) or GPU not supported

解决 / Solution
将conda环境所在文件夹中的cpp_extension.py内容从:

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'), ]) supported_arches = ['3.5', '3.7', '5.0', '5.2', '5.3', '6.0', '6.1', '6.2', '7.0', '7.2', '7.5']

改为

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'), ]) 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.6的支持, 3090就是属于sm86架构.
see solution: https://blog.csdn.net/ng323/a...)

> undefined symbol: THPVariableClass