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

猜你喜欢

转载自blog.csdn.net/weixin_44312422/article/details/130950637
今日推荐