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