配置和运行CenterNet时踩过的坑

%编程小白%

已于 2022-03-09 11:08:39 修改

阅读量3.8k

点赞数 3

文章标签： pytorch 人工智能 python

于 2022-03-09 11:07:14 首次发布

本文链接：https://blog.csdn.net/m0_49519243/article/details/123370065

版权

本文档详细记录了配置CenterNet环境时遇到的一系列问题，包括CUDA版本与torch不匹配、缺少用户权限、编码错误、依赖库错误和网络下载失败等。解决方法包括检查并匹配CUDA和torch版本、以管理员权限运行CMD、修改源码、正确粘贴DCNv2文件夹以及安装缺失的库。通过这些步骤，成功解决了所有问题，确保了CenterNet的顺利运行。

摘要由CSDN通过智能技术生成

配置CenterNet环境参考这篇博客：https://blog.csdn.net/startllee/article/details/118655091

问题一：No CUDA runtime is found

No CUDA runtime is found, using CUDA_HOME='C:\Program Files\NVIDIA GPU Computing ToolKit\CUDA\v10.2'

原因： cuda与torch的版本不匹配
解决：
1. 在cmd命令窗里输入pip3 list查看自己的torch版本
在这里插入图片描述
可以看到我的torch版本是1.9.1

2.在cmd命令窗输入nvcc --version查看自己的cuda版本号，我的是10.2

3.可以通过这个网址查看cuda匹配的版本https://pytorch.org/get-started/previous-versions/

4.为了和参考博客的版本保持一致，我去下载了cuda10.1版本，可以参考这篇博客https://blog.csdn.net/u011473714/article/details/95042856

问题二：EnvironmentNotWritableError

原因： 用户权限不够，cmd需要用管理员权限运行
解决： 打开c盘——》Windows文件夹——》System32文件夹——》找cmd.exe——》右键点击管理员运行
在这里插入图片描述
到这儿三个任务都done就ok了。

问题三：UserWarning

在这里插入图片描述
问题： 编译器错误
解决： 按照报错路径搜索打开cpp_extendion.py，使用ctrl+f搜索match,匹配到源码的第205行左右，将compiler_info.decode()改为compiler_info.decode(' gbk')
在这里注意，我刚开始是在源码基础上改的，结果还是报错了，后来把源码注释掉，回车重写一行就可以了。
在这里插入图片描述

问题四：编译出现error:commond

在这里插入图片描述
原因： 之间把下载好的DCNv2粘到了CnterNet路径下
解决： 将CenterNet里面的DCNv2先删除掉，接着再将下载好的DCNv2粘过去

问题五：ModuleNotFoundError

在这里插入图片描述
问题： 没有progress组件
解决： 使用pip3 install progress进行安装

问题六：urllib.error.HTTPError

这个错误忘记截图了，报错部分完整是错误：urllib.error.HTTPError: HTTP Error 404: Not Found
原因： 网络超时，测试时没有用到本地下载好的模型，系统选择自己联网重新下载
解决： 进入E:\CenterNet\src\lib\models\networks路径下，打开pose_dla_dcn.py文件，注释掉注释313和314行即可
在这里插入图片描述