- 需要确认您的 Windows 7/8/10 是 64 位操作系统*
- 需要您具有支持 CUDA 的 nVidia 显卡,且正确安装 CUDA 10
注意:Windows 仅支持 CUDA 8.0/9.0/10.0 的单卡模式;不支持 CUDA 9.1/9.2/10.1 - 需要使用 cuDNN 7.3+
- Windows 暂不支持 NCCL
- 确认您需要安装 PaddlePaddle 的 Python 是您预期的位置,因为您计算机可能有多个 Python
where python3
(根据您的环境您可能需要替换本说明中所有命令行的 python3
为python
,或者替换为具体的 Python 路径)
- 同时检查 Python3 的版本,确认是 3.5.1+/3.6/3.7:
python3 --version
- 确认 Python 有对应的 pip,检查 Python 对应的 pip 的版本,确认是 9.0.1+:
python3 -m ensurepip
python3 -m pip --version
- 确认 Python 和 pip 是 64 bit,并且处理器架构是x86_64(或称作 x64、Intel 64、AMD64)架构,目前PaddlePaddle不支持arm64架构
下面的第一行输出的是 "64bit",第二行输出的是 "x86_64"、"x64" 或 "AMD64" 即可:
python3 -c "import platform;print(platform.architecture()[0]);print(platform.machine())"
python3 -m pip install paddlepaddle-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple
或
python3 -m pip install paddlepaddle-gpu==1.6.2.post107 -i https://pypi.tuna.tsinghua.edu.cn/simple