- 环境准备
为了让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的版本的对应。