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.
Original: https://www.cnblogs.com/cnhemiya/p/16101905.html
Author: 小熊宝宝啊
Title: Ubuntu百度飞桨和CUDA的安装
原创文章受到原创版权保护。转载请注明出处：https://www.johngo689.com/711698/
转载文章受原作者版权保护。转载请注明原作者出处！