在jetson-Nano上安装pycuda失败怎么办
对于使用jetson-Nano的同学,在学习和使用的过程中 ,不知道又没用用到过pycuda这个库,本篇文章就是记录一下本人在使用jetson-Nano做车道线检测的时候,安装pycuda遇到的问题,以及是如何解决的
pycuda是干什么的
CUDA是一个显卡编程语言,而pycuda就是一个方便的、不需要学习CUDA语言又能够进行显卡编程的python模块。
安装pycuda遇到的问题
主要错误代码
error: command 'aarch64-linux-gnu-gcc' failed with exit status 1
原因分析
个人觉得可能的原因也就两个:
- 安装pycuda时需要安装相应的依赖库
- CUDA的环境变量没有设置
- (觉得不对的同学,轻点喷,以上只是个人观点,仅供参考~~~)
如何解决
first
修改文件
sudo vim ~/.bashrc
输入环境变量的地址,在最后直接复制粘贴就好。。
second
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
export CUDA_ROOT=/usr/local/cuda
thirty
source ~/.bashrc
大家记得修改之后一定要source一下,不然之前白改动了。
source之后就可以看到自己的cuda版本了
请输入以下指令查看:
nvcc - V
four
pip3 install pycuda==2020.1
pycuda的安装也是要注意版本对应的,主要是看python和cuda两个,不过现在大家在jetson-Nano中安装的python应该是3.6以上,cuda是10.0以上的版本了,所以就直接安装2020.1就好了。。。(当然这也是截止到我写这篇帖子的时间,以后的你看到这篇文章的时候,还是要注意一下,看看对应的版本匹配!!)
five
安装成功!!
THE END
本次记录到此结束,下次再来!!~~