笔者近期想使用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之后,可以进行简单测试以便于确认是否安装成功