CenterNet环境配置问题之fatal error: cuda.h: 没有那个文件或目录

CenterNet环境配置问题之fatal error: cuda.h: 没有那个文件或目录

• 我的环境ubuntu16.04;CUDA 9.0;python 3.6;

• 我跟着作者的install.md文档配置环境时，我跟着文档下载了pytorch 0.4.1

• 当我进入路径： ~/CenterNet/src/lib/models/networks/DCNv2，运行mash.sh脚本时，出现了以下问题

...
...fatal error: cuda.h: 没有那个文件或目录
...
...
cffi.VerificationError: CompileError: command 'gcc' failed with exit status 1

• 试过了很多的方法，总的来说就是文件夹DCNv2与pytorch, CUDA版本的匹配.

• 最后我的电脑最后匹配的组合时DCNv2(不同与原来项目中的文件夹)，pytorch 1.1.0；CUDA 9.0

• 进入pytorch官网

• 找到适合自己环境的pytorch版本命令行上运行下载命令，我的环境匹配的下载命令是：

conda install pytorch torchvision cudatoolkit=9.0 -c pytorch

• 再次运行mash.sh脚本，问题解决

• 参考

  https://github.com/xingyizhou/CenterNet/issues/9

  https://github.com/locuslab/pytorch_fft/issues/21

  https://github.com/horovod/horovod/issues/383