谷歌colab上安装百度paddlepaddle框架

时间:2022-07-23
本文章向大家介绍谷歌colab上安装百度paddlepaddle框架,主要内容包括其使用实例、应用技巧、基本知识点总结和需要注意事项,具有一定的参考价值,需要的朋友可以参考一下。

安装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了。