最近需要在 Ubuntu 20.04 环境下安装飞桨 Paddle 3.2,在实际操作过程中遇到了一些问题。为了帮助大家避免踩坑,本文将详细记录我的解决过程,并提供一些安装优化建议。
环境准备
首先,确保你的 Ubuntu 20.04 系统已经安装了必要的依赖包。
确认 Python 版本
飞桨 Paddle 通常需要 Python 3.7 或更高版本。可以通过以下命令检查 Python 版本:
python3 --version
如果 Python 版本过低,需要进行升级。可以使用 apt 包管理器安装 Python 3.7 或 3.8:
sudo apt update
sudo apt install python3.8 python3-pip # 安装 python3.8 和 pip
sudo update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.8 1 # 设置 python3 默认版本
python3 --version # 再次验证
安装 pip
确保 pip 已正确安装并更新到最新版本:
sudo apt install python3-pip
pip3 install --upgrade pip
安装飞桨 Paddle 3.2
官方推荐使用 pip 安装飞桨 Paddle。
CPU 版本安装
如果不需要 GPU 支持,可以直接安装 CPU 版本:
pip3 install paddlepaddle==3.2 -i https://mirror.baidu.com/pypi/simple
国内用户建议使用百度镜像源,可以加快下载速度。类似于使用阿里云或者清华的镜像源。
GPU 版本安装(CUDA)
如果要使用 GPU 版本,需要确保已经安装了 NVIDIA 驱动、CUDA 和 cuDNN。 这是安装飞桨 Paddle 的一个非常常见的踩坑点。
安装 NVIDIA 驱动
可以从 NVIDIA 官网下载对应 Ubuntu 20.04 的驱动,或者使用 apt 安装:
sudo apt update
sudo apt install nvidia-driver-510 # 示例,根据你的显卡型号选择合适的驱动版本
nvidia-smi # 验证驱动是否安装成功
安装 CUDA 和 cuDNN
从 NVIDIA 官网下载 CUDA Toolkit 和 cuDNN,并按照官方文档进行安装。安装过程中,注意配置环境变量。
通常需要设置以下环境变量:
export CUDA_HOME=/usr/local/cuda
export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH
export PATH=$CUDA_HOME/bin:$PATH
将这些环境变量添加到 ~/.bashrc 文件中,并执行 source ~/.bashrc 使其生效。
安装 GPU 版本 Paddle
安装 GPU 版本的飞桨 Paddle:
pip3 install paddlepaddle-gpu==3.2 -i https://mirror.baidu.com/pypi/simple
验证安装
安装完成后,可以通过以下代码验证是否安装成功:
import paddle
paddle.utils.run_check()
如果输出 Running verify PaddlePaddle program successfully!,则表示安装成功。
常见问题与解决方案
问题 1:ImportError: libcudart.so.11.0: cannot open shared object file
原因: CUDA 库路径未正确配置。
解决方案: 检查 LD_LIBRARY_PATH 环境变量是否包含 CUDA 库路径,并确保 CUDA 版本与 Paddle 版本兼容。
问题 2:安装 Paddle 时出现依赖冲突
原因: 系统中已安装的其他软件包与 Paddle 的依赖冲突。
解决方案: 尝试使用 venv 创建一个独立的 Python 虚拟环境,并在虚拟环境中安装 Paddle。 这也是我个人非常推荐的做法,避免系统环境污染。
python3 -m venv myenv
source myenv/bin/activate
pip3 install paddlepaddle==3.2 -i https://mirror.baidu.com/pypi/simple # 或 paddlepaddle-gpu
问题 3:GPU 运行速度慢
原因: 可能没有正确使用 GPU,或者 GPU 驱动版本过低。
解决方案: 检查 CUDA、cuDNN 和 NVIDIA 驱动版本是否匹配,并确保 Paddle 可以检测到 GPU 设备。可以尝试更新驱动到最新版本。
优化建议
- 使用国内镜像源: 可以显著提高下载速度。
- 使用虚拟环境: 避免依赖冲突。
- 及时更新依赖包: 保持环境的最新状态。
希望本文能帮助你顺利在 Ubuntu 20.04 上安装飞桨 Paddle 3.2。 祝大家顺利!
冠军资讯
代码一只喵