Win10 下安装 PaddlePaddle

原创已于 2022-03-20 00:17:40 修改 · 953 阅读

9 ·

CC 4.0 BY-SA版权

文章标签：

#windows 10 #paddlepaddle

于 2021-07-09 23:49:27 首次发布

装机专栏收录该内容

8 篇文章

订阅专栏

本文记录了一位用户安装CUDA 10.2、cuDNN 7.65以及PaddlePaddle的过程。首先，用户通过NVIDIA控制面板查看CUDA兼容版本，然后从NVIDIA官网下载并安装CUDA。安装时，用户选择了自定义选项，仅安装CUDA核心组件。接着，为配合CUDA 10.2，用户安装了cuDNN 7.65，并将库文件与CUDA相应目录合并。最后，用户通过conda在虚拟环境中安装了PaddlePaddle 2.1.1，并运行测试确认安装成功。

部署运行你感兴趣的模型镜像

记录下自己安装 PaddlePaddle 的血泪史。
参考 PaddlePaddle官网

准备工作

先查看自己 NVIDIA 所支持的 CUDA 版本。

打开NVIDIA控制面板
查看系统信息，选择组件

安装 CUDA

我这里安装的是 cuda 10.2，因为 paddle 官方推荐非安培架构使用10.2性能更优。
我没看懂 CUDA 的官方文档，所以参考了别人的博客。

附上 CUDA 的安装网址：https://developer.nvidia.com/cuda-toolkit-archive

下载安装

这里选择了本地安装。
在这里插入图片描述

安装时，我选择的是自定义。
在这里插入图片描述
然后只选择了CUDA，后面三个因为一般电脑都已经装好了这三个。

然后这个 Visual Studio Intergration 建议不要选。

测试

安装成功后，打开终端，输入nvcc -V测试是否安装成功。
在这里插入图片描述

安装 cuDNN

因为 CUDA 10.2 要求配合 cuDNN 的版本要大于 7.65。出于保险起见，我选择安装了 cuDNN 7.65。
在这里插入图片描述

根据官网文档，下载cuDNN，要先确保注册了NVIDIA开发人员计划。

这是 cuDNN 的安装网址：https://developer.nvidia.com/rdp/cudnn-download。

在这里插入图片描述
安装解压后，把bin、include、lib分别与 cuda 里的 bin、include、lib合并

安装 PaddlePaddle

我这边是采用官网的 conda 安装方式安装10.2。

# 打开 Anaconda Prompt，创建一个虚拟环境
conda create -n paddle_env python=3.8
# 进入虚拟环境
conda activate paddle_env
# 安装 paddlepaddle
conda install paddlepaddle-gpu==2.1.1 cudatoolkit=10.2 --channel https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/Paddle/

检验是否安装成功

还是在之前的虚拟环境中。

# 进入 python 解释器
python
# 导入 paddle
import paddle
paddle.utils.run_check()

看到 “PaddlePaddle is installed successfully!” 就表示安装成功了
在这里插入图片描述

您可能感兴趣的与本文相关的镜像

PyTorch 2.5

PyTorch

Cuda

PyTorch 是一个开源的 Python 机器学习库，基于 Torch 库，底层由 C++ 实现，应用于人工智能领域，如计算机视觉和自然语言处理