Cuda 程序编译报错: fatal error: cusparse.h: No such file or directory

编译cuda程序时发现下列报错：

/mnt/xxx/miniconda3/envs/xxx/lib/python3.8/site-packages/torch/include/ATen/cuda/CUDAContext.h:6:10: fatal error: cusparse.h: No such file or directory
#include <cusparse.h>
^~~~~~~~~~~~

检查发现是选择了错误的 Cuda 版本。

ls /usr/local/

发现有两个cuda版本:

cuda-10.2, cuda-11.1

目前系统里使用的是 cuda-10.2，因此使用下列指令指定后解决问题:

export CUDA_HOME=/usr/local/cuda-10.2

实测发现如果使用 export CUDA_HOME=/usr/local/cuda-11.1 就会发生上述报错。怀疑这个11.1的版本在安装的时候并不完全，导致缺失一些 .h 文件。