1. 安装必要的依赖项
sudo apt-get install libboost-all-dev
sudo apt-get install python-dev python-setuptoolslibboost-python-dev libboost-thread-dev -y
2. 下载软件包兵解压
首先选择与自己 cuda 对应版本的pycuda,本文已安装jetpack4.6,cuda10.2
到官方网址下载对应 软件包 pycuda-2021.1到/usr/local/目录下,并解压。
sudo tar xzvf pycuda-2021.1.tar.gz
cd /usr/local/pycuda-2021
3.安装
python3 configure.py --cuda-root=/usr/local/cuda-10.2
make -j4 # 解决各种的关键步骤,先编译
sudo python3 setup.py install
4. 验证
安装完成后,可以通过测试pycuda-2021.1里面的例子来验证安装:
cd /usr/local/pycuda-2021.1/examples
python3 demo.py

如果出现正确的 矩阵计算 结果,则表示PyCUDA安装成功。
Py CUDA安装 路径为/usr/local/pycuda-2021.1
点赞支持一下吧~💖💖💖
免责声明:本文系网络转载或改编,未找到原创作者,版权归原作者所有。如涉及版权,请联系删