ubuntu22.04安装pyCUDA

更多内容请查看 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之后，可以进行简单测试以便于确认是否安装成功，

​

​

​