CenterNet环境搭建记录

CenterNet仓库链接：

https://github.com/xingyizhou/CenterNet

记录一下跟着INSTALL.md走的过程遇到的一些问题

问题1：更改cudnn

这一步倒是没报错，记录一下这个命令的意思。

把对应py文件的1254行中的torch.backends.cudnn.enabled替换成False

执行完以后可以去py文件里检查一下，或者自己手动替换

# PYTORCH=/path/to/pytorch # usually ~/anaconda3/envs/CenterNet/lib/python3.6/site-packages/
# for pytorch v0.4.0
sed -i "1194s/torch\.backends\.cudnn\.enabled/False/g" ${PYTORCH}/torch/nn/functional.py
# for pytorch v0.4.1
sed -i "1254s/torch\.backends\.cudnn\.enabled/False/g" ${PYTORCH}/torch/nn/functional.py

问题2：make coco报错

gcc: error: pycocotools/_mask.c: No such file or directory
error: command 'gcc' failed with exit status 1
Makefile:3: recipe for target 'all' failed
make: *** [all] Error 1

原因是新环境没有安装cython， -->pip install cython即可

问题3：编译可形变卷积报错

/PycharmProjects/CenterNet/src/lib/models/networks/DCNv2$ ./make.sh

报错如下：

先是

./make.sh: line 5: nvcc: command not found
./make.sh: line 6: nvcc: command not found
./make.sh: line 9: nvcc: command not found
./make.sh: line 10: nvcc: command not found
然后

distutils.errors.DistutilsExecError: command 'gcc' failed with exit status 1

distutils.errors.LinkError: command 'gcc' failed with exit status 1

cffi.VerificationError: LinkError: command 'gcc' failed with exit status 1

distutils.errors.DistutilsExecError: command 'gcc' failed with exit status 1

首先解决nvcc 的命令未找到问题，需要nvcc命令添加到环境变量里面去

export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64
export PATH=$PATH:/usr/local/cuda-9.0/bin

根据自己的cuda路径修改，老版本的是lib文件夹，新版本是lib64，其他的都一样

然后再次运行make.sh，没有报错了

注意：电脑上可能有多个cuda版本，可以用nvcc -V查看当前终端的cuda版本，一定要和pytorch的要求对应起来，添加环境变量那里同理。

测试

上述几个问题解决了之后应该就安装成功了。按照教程把预训练模型下载好，移动到指定位置。

cd到src路径下，执行

python demo.py ctdet --demo ../images/ --load_model ../models/ctdet_coco_dla_2x.pth

成功！

CenterNet环境搭建记录相关推荐

TensorRT 环境搭建记录
TensorRT 环境搭建记录文章目录 TensorRT 环境搭建记录 1.TensorRT介绍 2.使用TensorRT总目标 3.Centos下TensorRT环境搭建(Ubuntu一样) 4. ...
极简darknet环境搭建记录
<极简darknet环境搭建记录> darknet的环境搭建已经非常简单了,但是偶尔的一点小坑,可能会浪费一些时间,比如编译的时候使用CUDA报错,或者没找到OpenCV-简单记录一下 ...
MIT Mini Cheetah–开源代码仿真环境搭建记录(Ubuntu20.04)
MIT Mini Cheetah–开源代码仿真环境搭建记录(Ubuntu20.04) 一．安装依赖包 sudo apt install mesa-common-dev freeglut3-dev co ...
Linux环境搭建记录——GitLab安装
Linux环境搭建记录 GitLab安装 GitLab安装更多GitLab详细信息请参考:GitLab官方文档分步指南获取GitLab资源可以访问 https://mirrors.tuna.t ...
Linux环境搭建记录——Jenkins安装
Linux环境搭建记录 Jenkins安装 Jenkins升级 Jenkins插件离线安装构件Jenkins项目 Jenkins安装更多Jenkins详细信息请参考:Jenkins官方文档分步指 ...
linux ubuntu安装pytorch（深度学习环境搭建记录，无sudo权限）踩坑全记录
一些牢骚:深度学习没怎么学习几次,搭建环境已经把我搞秃了哈哈哈. 之前在网上找到的搭建环境的步骤,我没有root权限,很多操作都不行(比如运行.run文件,cuda 和cudnn的安装和修改也需要ro ...
PX4环境搭建记录（ROS+Gazebo+mavros+PX4+QGC）
全过程记录PX4环境搭建(ROS+Gazebo+Mavros+PX4+QGC) 本人飞控专业在读,近段时间在老师的建议下,开始搭建PX4环境配置.因为并没有之前相关环境配置的经验,所以走了不少弯路,在 ...
Windows下Qt编写Android应用程序的环境搭建记录
目录前言配置开发环境安装JDK1.8 安装SDK 安装NDK Qt配置OpenCV C++&Java混合编程结语前言由于要在手机上实现一些应用,例如部署目标检测模型.添加单目测距定 ...
Conda环境搭建记录
背景: 虽然搭建一个conda环境的流程非常简单,但是,在虚拟机中配置或者换了一台设备总要再次搭建,再次搭建的时候有些命令就记不清了,基于此,写下此篇文章做个记录. 第一步.下载正确的安装包可以不去 ...

CenterNet环境搭建记录

问题1：更改cudnn

问题2：make coco报错

问题3：编译可形变卷积报错

测试

CenterNet环境搭建记录相关推荐

最新文章

热门文章