记录下自己安装 paddlepaddle 的血泪史。
参考 PaddlePaddle官网
准备工作
先查看自己 NVIDIA 所支持的 CUDA 版本。
- 打开NVIDIA控制面板
- 查看系统信息，选择组件
安装 CUDA
我这里安装的是 cuda 10.2，因为官方推荐非安培架构最好安装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合并
bin ->
安装 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!” 就表示安装成功了