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