错误示图:
使用sudo apt-get install nvidia-cuda-toolkit命令安装之后会自动安装一系列乱七八糟相关文件,然后出现以下显示
错误显示不是nvcc,而是nvcc’
解决办法同下述问题:
- python包编译过程中出现: /usr/local/cuda-9.0/nvcc: No such file or directory问题的简单解决办法:
export CUDA_HOME=/usr/local/cuda-10.0
pip install torch_sparse
踩过的其他坑(试过的其他可能有用的办法)
首先尝试在~/.bashrc中添加相关路径,如果不行,就在setup.py里添加
CUDAExtension(
'roi_align.crop_and_resize_gpu',
['roi_align/src/crop_and_resize_gpu.cpp',
'roi_align/src/cuda/crop_and_resize_kernel.cu'],
include_dirs=include_dirs,
extra_compile_args={'cxx': ['-g', '-fopenmp'],
'nvcc': ['-O2']}
)
注意:
修改环境变量后要用source ~/.bashrc保存以生效