pytorch对maskrcnn进行编译时遇到 fatal error: cuda.h: 没有那个文件或目录 #include “cuda.h“

1、按照mattnet上的步骤的第二步运行mattnet:

CUDA_VISIBLE_DEVICES=gpu_id python tools/extract_mrcn_head_feats.py --dataset refcoco --splitBy unc
CUDA_VISIBLE_DEVICES=gpu_id python tools/extract_mrcn_ann_feats.py --dataset refcoco --splitBy unc

2、提示no module named _ext，查找资料，应该是matt用到的maskrcnn/lib/里的文件需要编译，按照github上的maskrcnn的compile说明进行编译，cd进入lib文件夹，再执行

make 

3、报错：

发现是cuda的问题，nvidia-smi发现cuda存在，但是去user查找cuda路径不存在，最后安装cuda9.0，再次编译，发现编译成功。