谷歌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了。 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM