美文网首页
PaddlePaddle和GPU(一)

PaddlePaddle和GPU(一)

作者: sweetsky0901 | 来源:发表于2017-12-09 11:32 被阅读0次
  • 环境准备
    为了让paddlepaddle支持GPU模式,首先应该在你的机器上安装好GPU驱动和CUDA Toolkit。
    除此之外,还应该下载cuDNN。默认的CUDA安装包,是不包含cuDNN的。
    为了可以看到GPU设备和驱动的信息,在命令行执行

nvidia-smi

  • PaddlePaddle安装
    如果要使用GPU能力的PaddlePaddle,还需要在安装的时候,让PaddlePaddle支持GPU
    如果是pip安装的,则应该安装的是paddlepaddle_gpu
    如果是源码安装的,则应该在cmake的时候,带上编译选项-DWITH_GPU=ON

  • 注意事项
    需要注意的是,安装驱动的时候,首先要考虑到卡和驱动的版本的对应
    其次,要注意到PaddlePaddle编译的时候环境中的CUDA和cuDNN的版本和PaddlePaddle运行的环境中的CUDA和cuDNN的版本的对应。

相关文章

网友评论

      本文标题:PaddlePaddle和GPU(一)

      本文链接:https://www.haomeiwen.com/subject/moyuixtx.html