PaddlePaddle CPU/GPU安装

最新推荐文章于 2024-09-10 09:53:50 发布

来了就走下去

最新推荐文章于 2024-09-10 09:53:50 发布

阅读量2.2k

点赞数 1

CC 4.0 BY-SA版权

分类专栏： PaddlePaddle 文章标签： PaddlePaddle CUDA CuDNN GPU

本文链接：https://blog.youkuaiyun.com/u012009613/article/details/109578996

PaddlePaddle 专栏收录该内容

8 篇文章 ¥9.90 ¥99.00

订阅专栏

本文详细介绍了如何在Ubuntu16环境下安装PaddlePaddle的CPU和GPU版本。首先，强调了Python版本的选择，推荐使用3.7.2。接着，逐步讲解了安装GPU版本PaddlePaddle前的准备，包括CUDA 10.0的安装，CuDNN 7.6.5的匹配与下载，以及TensorRT 6.0.1的兼容性检查和下载。在安装过程中，提到了环境变量配置和可能出现的问题，如`import paddle.fluid`时的提示，解决方案是安装TensorRT。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1、安装需要的Python版本

注意：不要安装过高版本，安装支持的Python版本

2、选择安装CPU/GPU版本的PaddlePaddle

3、使用不同方式安装PaddlePaddle，这里使用pip安装

CPU版PaddlePaddle：
python -m pip install paddlepaddle -i https://mirror.baidu.com/pypi/simple（推荐使用百度源） 或 python -m pip install paddlepaddle -i https://pypi.tuna.tsinghua.edu.cn/simple
GPU版PaddlePaddle：
python -m pip install paddlepaddle-gpu -i https://mirror.baidu.com/pypi/simple 或 python -m pip install paddlepaddle-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple

我使用的Python版本是3.7.2，安装后出现如下提示：