在使用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