pycuda安装 cuda.h文件找不到问题

本文详细解析了PyCUDA安装中找不到CUDA.H文件的问题,提供了三种解决方案:添加CUDA.H搜索路径、配置环境变量CUDA_INC_DIR及正确使用pip权限安装PyCUDA。

对于安装pycuda安装 cuda.h文件找不到的问题,我之前安装tensorrt遇到过,不过最近我帮同事安装时也遇到了这个问题,但原来的方法无效,只好仔细分析问题,目前分析出来问题原因有三点。

(1)cuda.h这个文件的确找不到,只要添加cuda.h的搜索路径即可。

     https://blog.youkuaiyun.com/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,即可安装成功。

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值