利用源码编译方式:

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

成功!

Logo

欢迎来到由智源人工智能研究院发起的Triton中文社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐