安装paddlepaddle要求:
- 需要确认您的 Ubuntu 是 64 位操作系统*
- 需要您具有支持 CUDA 的 nVidia 显卡,且正确安装 CUDA 10
- Ubuntu 14.04 支持 CUDA 10.0/10.1
- Ubuntu 16.04 支持 CUDA 9.0/9.1/9.2/10.0/10.1
- Ubuntu 18.04 支持 CUDA 10.0/10.1
- 需要使用 cuDNN 7.3+
- 如果您需要多卡模式,请安装 NCCL 2,更多信息可见:PaddlePaddle 对 NCCL 支持情况
- 确认您需要安装 PaddlePaddle 的 Python 是您预期的位置,因为您计算机可能有多个 Python
1、查看cuda版本:paddlepaddle要求cuda为9或10
!cat /usr/local/cuda/version.txt
2、查看python版本和位置
输入:!which python查看python位置
输入:!python --version查看python版本
3、查看pip版本:确认 Python 有对应的 pip,检查 Python 对应的 pip 的版本,确认是 9.0.1+:
!python -m pip --version
4、确认 Python 和 pip 是 64 bit,并且处理器架构是x86_64架构,目前PaddlePaddle不支持arm64架构
下面的两个命令分别输出的是 "64bit" 和 "x86_64" 即可
!python -c "import platform;print(platform.architecture()[0]);print(platform.machine())"
5、安装paddlepaddle
!python -m pip install paddlepaddle-gpu==1.7.0.post107 -i https://mirror.baidu.com/pypi/simple
6、验证是否安装成功
至此,大功告成,可以使用paddlepaddle了。