Ubuntu22.04安装paddle GPU版本

文章目录

- - 确立版本
  - 安装CUDA与CUDNN
  - 安装paddle

确立版本

查看官网信息，确立服务版本：https://www.paddlepaddle.org.cn/documentation/docs/zh/2.6/install/pip/linux-pip.html

安装CUDA与CUDNN

通过nvidia-smi查看当前显卡驱动版本：
通过https://docs.nvidia.cn/cuda/cuda-toolkit-release-notes/index.html找到与之对应的 CUDA 版本：
再通过https://developer.nvidia.com/cuda-toolkit-archive/下载该版本的 CUDA ：
```
wget https://developer.download.nvidia.com/compute/cuda/12.0.0/local_installers/cuda_12.0.0_525.60.13_linux.run
sudo sh cuda_12.0.0_525.60.13_linux.run --override
```
报的警告选择 continue ：

根据提示选择安装选项，此处我使用的是lxd容器，之前已经安装过显卡驱动，所以此处选择的不安装，如果是物理机或者第一次配置环境，此处应勾选：

安装完成后提示：

根据上图提示需要配置环境变量：

vim  ~/.bashrc

再文件最后加入以下语句（加入的具体内容根据上述输出进行修改）：

export CUDA_HOME=/usr/local/cuda-12.0
export LD_LIBRARY_PATH=${CUDA_HOME}/lib64
export PATH=${CUDA_HOME}/bin:${PATH}

然后使其生效：

source ~/.bashrc

可以使用命令 nvcc -V 查看安装的版本信息：
进入https://developer.nvidia.cn/rdp/cudnn-archive下载对应 CUDA 的 cuDNN ：

下载好解压：

tar xvf cudnn-linux-x86_64-8.9.1.23_cuda12-archive.tar.xz

进入解压后的目录运行以下命令：

sudo cp include/cudnn*.h  /usr/local/cuda/include
sudo cp lib/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn*.h 
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

拷贝完成之后，可以使用以下命令查看CUDNN的版本信息：
```
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
```
表示版本信息为8.9.1。

安装paddle

查看文档https://www.paddlepaddle.org.cn/documentation/docs/zh/2.6/install/pip/linux-pip.html，使用对应命令安装 paddle：
```
python3 -m pip install paddlepaddle-gpu==2.6.1.post120 -f https://www.paddlepaddle.org.cn/whl/linux/mkl/avx/stable.html
```