cusparse.h: No such file or directory compilation terminated.

在使用python setup.py build develop安装awesome-semantic-segmentation-pytorch碰到了如下问题:

In file included from /xxx/awesome-semantic-segmentation-pytorch/core/nn/csrc/cuda/syncbn_cuda.cu:5:0:
/usr/local/lib/python3.5/dist-packages/torch/lib/include/ATen/cuda/CUDAContext.h:12:22: fatal error: cusparse.h: No such file or directory
compilation terminated.
error: command '/usr/local/cuda/bin/nvcc' failed with exit status 1

实验环境:

  • python 3.5
  • pytorch 1.1.0
  • CUDA 9.0

问题分析

这是因为cuda中缺少cusparse.h这个文件,我得解决方法是:
发现使用的cuda command错了,应该是'/usr/local/cuda-9.0/bin/nvcc,原来是因为/usr/local下有一个cuda和一个cuda-9.0文件夹,把cuda移走,使用正确的cuda在运行就没有问题了。
其他可能可行的解决方案:(若原因跟我不同)

  • 将缺少的cusparse.h下载放入对应目录
  • 重新安装CUDA

你可能感兴趣的:(pytorch)