jetson 安装 pycuda
·
利用源码编译方式：
pycuda 官方下载地址https://pypi.org/project/pycuda/#history
1. 下载相应压缩包，我选择的最新版本，解压：
sudo tar xvf pycuda-2023.1.tar.gz
进入解压后的文件夹中：
cd pycuda-2023.1
2. 看一下cuda 是否运行正常，利用下面命令：
nvcc --version
查看cuda 版本，若出现版本和位置，则继续；如果报错就需要配置一下cuda 的环境变量:
进行cuda的环境配置：
sudo gedit ~/.bashrc
打开文件后，在最后添加如下信息：（根据自己cuda版本进行编辑）
export PATH="/usr/local/cuda-11.4/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-11.4/lib64:$LD_LIBRARY_PATH"
关闭文件，使其生效：
source ~/.bashrc
最后再输入nvcc --version即可查看nvcc版本，成功就可以进行第三步了:
3. 依次执行下面命令：
python3 configure.py --cuda-root=/usr/local/cuda-11.4
# 这里cuda-11.4 是你的cuda 文件夹名称，需要按照你的版本修改
make -j4
sudo python3 setup.py install
安装完成后，可以运行pycuda 提供的示例进行测试：
cd examples
python3 demo.py
成功！
北京人形旗下天工造物具身智能开源社区，聚焦具身天工与慧思开物两大平台
更多推荐
5
0- 0
已为社区贡献3条内容
所有评论(0)