利用源码编译方式:
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
成功!