jetson Xavier 安装pycuda 报错 Failed building wheel for pycuda_error: failed building wheel for pycuda

报错信息
note: This error originates from a subprocess, and is likely not a problem with pip.
ERROR: Failed building wheel for pycuda
Failed to build pycuda
ERROR: Could not build wheels for pycuda, which is required to install pyproject.toml-based projects

最简单的方法

新建一个虚拟环境，按nvidia官方给的方法装

conda create -n py310 python=3.10
conda activate py310
git clone https://github.com/NVIDIA/TensorRT.git
cd TensorRT/
python3 -m pip install -r requirements.txt
1
2
3
4
5

pytorch、torchvision、pycuda、onnx、onnxruntime等等都会一条命令装好，非常方便。

如果以上不满足你的开发需求，尝试以下方法

解决方法一
1、确定自己的cuda以及python版本
点此查看对应pycuda版本
注：cuda114代表cuda=11.4 ，cp38代表python=3.8，对应pycuda版本为2021.1。

2、修改~/.bashrc

sudo su
1

vim ~/.bashrc
1

export PATH=/usr/local/cuda-11.4/bin:/usr/local/cuda/bin:$PATH
1

上面的cuda-11.4记得改成自己的cuda版本

source ~/.bashrc
1

nvcc -V 
1

看一下有没有成功输出cuda版本信息

2、安装

sudo apt-get install python3-pip
pip3 install Cython
sudo apt-get update
sudo apt-get install -y build-essential libatlas-base-dev
sudo apt-get install libatlas-base-dev gfortran
pip3 install pycuda==2021.1
1
2
3
4
5
6

2021.1记得改成跟自己环境匹配的版本

解决方法二
如果以上pip安装方法不行：
最后一步的
pip3 install pycuda==2021.1
改为：
1、下载源码
2、解压安装

tar -zxvf pycuda-2021.1.tar.gz
cd pycuda-2021.1 
export PATH=/usr/local/cuda/bin:$PATH
python3 setup.py build
sudo python3 setup.py install
1
2
3
4
5