centos 7.5系统安装了cuda9.0,使用tensorRT进行模型优化,需要安装pycuda 出现的问题
首先查看官网,安装方式为pip install pycuda 报错
gcc: error trying to exec 'cc1plus': execvp: No such file or directory
执行yum install gcc-c++后,问题解决.
又报错:
src/cpp/cuda.hpp:14:18: fatal error: cuda.h: No such file or directory
#include <cuda.h>
^
解决方法 在/etc/profile.d/创建一个cuda.sh,添加下面代码:
export PATH=/usr/local/cuda-9.0/bin:$PATH
退出环境,然后运行sudo su 运行
重新source activate *****(自己环境)
从网上查看,有人说pycuda对应的cuda版本没有9.0的,可是我记得是可以装的,而且我没有从官网上查到其对应的版本必须要cuda8.0或者是cuda10.0的,所以我就运行了pip install pycuda-2019.1,成功了,哈哈·····,搞了几个小时,终于搞好了