Ubuntu22.04安装paddle GPU版本

Dragon水魅2025-01-20 20:48

文章目录

确立版本

查看官网信息，确立服务版本：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 ：
bash 复制代码
```
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容器，之前已经安装过显卡驱动，所以此处选择的不安装，如果是物理机或者第一次配置环境，此处应勾选：

安装完成后提示：
根据上图提示需要配置环境变量：
bash 复制代码
```
vim  ~/.bashrc
```
再文件最后加入以下语句（加入的具体内容根据上述输出进行修改）：
bash 复制代码
```
export CUDA_HOME=/usr/local/cuda-12.0
export LD_LIBRARY_PATH=${CUDA_HOME}/lib64
export PATH=${CUDA_HOME}/bin:${PATH}
```
然后使其生效：
bash 复制代码
```
source ~/.bashrc
```
可以使用命令 nvcc -V 查看安装的版本信息：
进入https://developer.nvidia.cn/rdp/cudnn-archive下载对应 CUDA 的 cuDNN ：

下载好解压：

bash 复制代码

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

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

bash 复制代码

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的版本信息：
bash 复制代码
```
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：
bash 复制代码
```
python3 -m pip install paddlepaddle-gpu==2.6.1.post120 -f https://www.paddlepaddle.org.cn/whl/linux/mkl/avx/stable.html
```