Jetson安装pycuda通用教程

1.pip 安装方式

直接使用pip安装方式：

pip3 install projects  # 先提前安装个projects
pip3 install pycuda==2021.1 -i https://mirror.baidu.com/pypi/simple

这里安装的2021.1版本，使用https://mirror.baidu.com/pypi/simple源安装，这种方式也可以安装成功，不过需要等待时间，如果pip方式安装失败，那就只能源码编译

2.源码编译方式

pycuda 官方下载地址https://pypi.org/project/pycuda/#history

以2021.1版本为例

下载后压缩包，解压：

sudo tar xvf  pycuda-2021.1.tar.gz

然后进入解压后的文件夹中：

cd pycuda-2021.1

先看一下cuda 是否运行正常， nvcc -V，查看cuda 版本，若出现下面界面，则继续，否则需要配置一下cuda 的环境变量：

依次执行下面命令：

python3 configure.py --cuda-root=/usr/local/cuda-11.4 # 1.这里cuda-11.4 是你的cuda 文件夹名称，需要按照你的版本修改
make -j4  # 2.
sudo python3 setup.py install  # 3.

安装完成后，可以运行pycuda 提供的示例进行测试：

cd  examples
python3 demo.py