Ubuntu 百度飞桨和 CUDA 的安装

Ubuntu 百度飞桨 和 CUDA 的安装

1、简介

本文主要是 Ubuntu 百度飞桨 和 CUDA 的安装

系统：Ubuntu 20.04

百度飞桨：2.2 为例

2、百度飞桨安装

访问百度飞桨 进入“安装”

版本：2.2 稳定版

操作系统：Linux

安装方式：pip

计算平台：CUDA11.2

如果你想安装CPU版本计算平台选择 CPU

# 百度 paddlepaddle gpu cuda 11.2
python3 -m pip install paddlepaddle-gpu==2.2.2.post112 -f https://www.paddlepaddle.org.cn/whl/linux/mkl/avx/stable.html

# 百度 paddlepaddle cpu
python3 -m pip install paddlepaddle==2.2.2 -i https://mirror.baidu.com/pypi/simple

3、安装 NVIDIA CUDA

需要管理员权限

# 添加 nvidia cuda 源
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600sudo
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/7fa2af80.pub
sudo apt-add-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/ /"

# 更新源
sudo apt update

# 安装 cuda
sudo apt -y install cuda

# 安装 cudnn
suda apt -y install libcudnn8 libcudnn8-dev

4、验证是否安装成功

python3 -c "import paddle;paddle.utils.run_check()"

如果安装成功会会显示类似于下面的结果

Running verify PaddlePaddle program ...
W0405 12:44:28.669028 16588 device_context.cc:447] Please NOTE: device: 0, GPU Compute Capability: 6.1, Driver API Version: 11.6, Runtime API Version: 11.2
W0405 12:44:28.673214 16588 device_context.cc:465] device: 0, cuDNN Version: 8.4.
PaddlePaddle works well on 1 GPU.
PaddlePaddle works well on 1 GPUs.
PaddlePaddle is installed successfully! Let's start deep learning with PaddlePaddle now.