測試下來最快速的步驟:
第一步,用pip install安裝paddlepaddle-gpu
python -m pip install paddlepaddle-gpu -i https://mirror.baidu.com/pypi/simple
該命令對應的cudatoolkit=10.2, cudnn=7.6.5
第二步,用conda install安裝cudatoolkit和cudnn
conda install cudatoolkit=10.2.89 cudnn=7.6.5
第三步,驗證飛槳安裝
python -c "import paddle;paddle.utils.run_check()"
獲得下圖,則安裝成功
第四步,用pip install安裝paddlex
pip install paddlex -i https://mirror.baidu.com/pypi/simple
第五步,用pip install安裝pycocotools
pip install pycocotools -i https://mirror.baidu.com/pypi/simple
第六步,驗證PaddleX安裝
python -c "import paddlex as pdx;print(pdx.__version__)"
獲得下圖,則安裝成功