- 環(huán)境準備
為了讓paddlepaddle支持GPU模式，首先應(yīng)該在你的機器上安裝好GPU驅(qū)動和CUDA Toolkit腹备。
除此之外谭贪，還應(yīng)該下載cuDNN。默認的CUDA安裝包令花，是不包含cuDNN的阻桅。
為了可以看到GPU設(shè)備和驅(qū)動的信息，在命令行執(zhí)行
nvidia-smi
PaddlePaddle安裝
如果要使用GPU能力的PaddlePaddle兼都，還需要在安裝的時候嫂沉，讓PaddlePaddle支持GPU
如果是pip安裝的，則應(yīng)該安裝的是paddlepaddle_gpu
如果是源碼安裝的俯抖，則應(yīng)該在cmake的時候输瓜，帶上編譯選項-DWITH_GPU=ON注意事項
需要注意的是，安裝驅(qū)動的時候芬萍，首先要考慮到卡和驅(qū)動的版本的對應(yīng)
其次尤揣，要注意到PaddlePaddle編譯的時候環(huán)境中的CUDA和cuDNN的版本和PaddlePaddle運行的環(huán)境中的CUDA和cuDNN的版本的對應(yīng)。
