飞桨学习笔记之安装 paddlepaddle-gpu

(需要提前安装VS)
下载 CUDA 10.0:https://developer.nvidia.com/cuda-10.0-download-archive
飞桨学习笔记之安装 paddlepaddle-gpu
安装完成后验证。
飞桨学习笔记之安装 paddlepaddle-gpu
下载 cuDNN:(需要先注册)
https://developer.nvidia.com/cudnn-download-survey
解压后,将3个文件分别复制到 C 盘 CUDA 对应的文件夹里。

查看 CUDA 版本:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA 。
查看 cuDNN 版本:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\include,找到名为 cudnn.h 的文件,打开看到代码中的前部声明部分。

确认 paddlepaddle-gpu 的安装环境:
飞桨学习笔记之安装 paddlepaddle-gpu
python 版本是3.5.1+/3.6+/3.7+:
python --version

确认 pip 的版本是 9.0.1+:
python -m ensurepip
python -m pip --version

确认 Python 和 pip 是 64 bit,并且处理器架构是x86_64(或称作 x64、Intel 64、AMD64)架构:
python -c “import platform;print(platform.architecture()[0]);print(platform.machine())”

可能需要安装 mkl:
pip install mkl

安装 paddlepaddle-gpu:
python -m pip install paddlepaddle-gpu -i https://mirror.baidu.com/pypi/simple

验证:
使用 python 进入 python 解释器,输入 import paddle.fluid ,再输入 paddle.fluid.install_check.run_check()。
飞桨学习笔记之安装 paddlepaddle-gpu