ubuntu22.04安装pyCUDA

最新推荐文章于 2024-06-06 14:49:26 发布

dataat

最新推荐文章于 2024-06-06 14:49:26 发布

阅读量1.2k

点赞数 1

分类专栏： python 文章标签： python pycuda

本文链接：https://blog.csdn.net/booklijian/article/details/130228231

版权

python 专栏收录该内容

46 篇文章 1 订阅

订阅专栏

更多内容请查看 www.laowubiji.com

笔者近期想使用GPU进行并行计算，搜索之后看到需要用到pyCUDA库函数，所以需要在所使用的ubuntu22.04系统中部署pyCUDA库，没想到在部署过程中折腾了好几回，总算是安装成功了。简单记录过程如下：

第一步：安装nvidia驱动

ubuntu-drives devices

根据上述命令选择适当的 nvidia驱动，例如笔者的NVIDIA GeForce RTX 2060 SUPER，显示信息如下：

== /sys/devices/pci0000:00/0000:00:03.0/0000:02:00.0 ==
modalias : pci:v000010DEd00001F06sv00001B4Csd000013A3bc03sc00i00
vendor   : NVIDIA Corporation
model    : TU106 [GeForce RTX 2060 SUPER]
driver   : nvidia-driver-470-server - distro non-free
driver   : nvidia-driver-510 - distro non-free
driver   : nvidia-driver-450-server - distro non-free
driver   : nvidia-driver-515-server - distro non-free
driver   : nvidia-driver-525-open - distro non-free recommended
driver   : nvidia-driver-515 - distro non-free
driver   : nvidia-driver-525-server - distro non-free
driver   : nvidia-driver-525 - third-party non-free
driver   : nvidia-driver-530 - third-party non-free
driver   : nvidia-driver-515-open - distro non-free
driver   : nvidia-driver-470 - distro non-free
driver   : xserver-xorg-video-nouveau - distro free builtin

笔者选择其中的nvidia-driver-525，因此使用如下命令进行安装

apt install nvidia-driver-525 即可成功安装

重启之后，测试一下 nvidia-smi，可以看到显卡相关信息则表示安装成功。

第二步安装cuda驱动

https://developer.nvidia.com/cuda-downloads

在该页面中按照提示依次选择定制下载软件包，笔者选择如下图所示：

其中给出的命令行如下，直接复制运行即可。

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-ubuntu2204.pin
sudo mv cuda-ubuntu2204.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/12.1.0/local_installers/cuda-repo-ubuntu2204-12-1-local_12.1.0-530.30.02-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2204-12-1-local_12.1.0-530.30.02-1_amd64.deb
sudo cp /var/cuda-repo-ubuntu2204-12-1-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cuda

第三步：安装pycuda

这一步笔者折腾了好几次，每次都是安装报错或者是安装完之后发现nvdia-smi无法运行，多次纠结之后，笔者使用的是：

pip3 install pycuda

而如果使用apt install python3-pycuda则可以正常指定，但是执行之后nvdia-smi无法运行。

另外笔者还看到一篇文章提到

pip install cuda-python

但是有可能版本不一致导致出问题，所以有的时候需要指定版本，例如：

pip install cuda-python==11.7

其中版本需要根据实际情况决定

安装了pycuda之后，可以进行简单测试以便于确认是否安装成功，