解决报错ModuleNotFoundError: No module named ‘nms.cpu_nms‘

在linux中跑项目的时候遇到了这个问题

原因是我的conda环境的python版本是3.9，而nms相关文件是python3.8编译得到的，如图

因此需要重新执行一下setup.py文件来生成对应版本的文件

python3 setup.py build_ext --inplace

但是在运行这个之后有个问题。

我第一次运行的时候只生成了

没有生成GPU的，并且有报错信息 nvcc fatal: Value 'sm_35' is not defined for option 'gpu-architecture'

是因为 sm_35（即 CUDA 的架构标识符）不受当前 CUDA 版本（CUDA 12.1）的支持。sm_35 是旧版本的架构，较新的 CUDA 版本可能已经不再支持它。

要解决这个问题，这时候要到setup.py文件中修改一下。可以尝试将 sm_35 替换为适合你的 GPU 的架构标识符。

GPU 的架构可以通过型号来确定，例如：

• Tesla V100、P100、T4 属于 Volta 架构（使用 sm_70）。
• RTX 20xx 系列（如 RTX 2080）和 Tesla T4 属于 Turing 架构（使用 sm_75）。
• RTX 30xx 系列（如 RTX 3080、A100） 属于 Ampere 架构（使用 sm_80 或 sm_86）。
• RTX 40xx 系列（如 RTX 4090） 属于 Ada 架构（使用 sm_89 或 sm_90，需更高 CUDA 支持）。

我的是RTX 3090，因此将sm_35改为sm_80，再执行命令，就会生成了符合版本的cpu和gpu文件

python3 setup.py build_ext --inplace

 

 就把3.8版本的删掉了，报错就解决了。