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.