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版本命令行上运行下载命令,我的环境匹配的下载命令是:
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch