pycuda安装 cuda.h文件找不到问题
对于安装pycuda安装 cuda.h文件找不到的问题,我之前安装tensorrt遇到过,不过最近我帮同事安装时也遇到了这个问题,但原来的方法无效,只好仔细分析问题,目前分析出来问题原因有三点。
(1)cuda.h这个文件的确找不到,只要添加cuda.h的搜索路径即可。
https://blog.csdn.net/u011337602/article/details/81238164
(2)未添加CUDA_INC_DIR到/etc/profile
在/etc/profile最后两行添加 export CUDA_INC_DIR=$CUDA_INC_DIR:/usr/local/cuda-9.0/include
保存并执行source /etc/profile
(3)安装pycuda使用的是sudo pip3 install pycuda
这种情况下,执行的root权限,但由于pycuda安装时会使用nvcc,而nvcc是不具有root权限的,所以会导致安装失败。
因此需要去掉sudo,只在用户权限下执行pip3 install pycuda,即可安装成功。
本文来自互联网用户投稿,文章观点仅代表作者本人,不代表本站立场,不承担相关法律责任。如若转载,请注明出处。 如若内容造成侵权/违法违规/事实不符,请点击【内容举报】进行投诉反馈!
