记录下自己安装 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。
conda create -n paddle_env python=3.8
conda activate paddle_env
conda install paddlepaddle-gpu==2.1.1 cudatoolkit=10.2 --channel https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/Paddle/
检验是否安装成功
还是在之前的虚拟环境中。
python
import paddle
paddle.utils.run_check()
看到 “PaddlePaddle is installed successfully!” 就表示安装成功了
|