安装

  • 1、安装paddle
  • 2、安装nccl
  • 3、验证

这次安装主要使用conda,可以有更好的安装体验。关于框架,常用的tensorflow,pytorch,但是国产的paddle也做的越来越好,而且学习资源有很多,我们应该抛开框架本身,更多关注算法。再说,paddle和pytorch很个似,确实需要某一个固定的框架,我觉的可以转写一下。

1、安装paddle

按照正常官网来操作即可,

conda install paddlepaddle-gpu==2.3.2 cudatoolkit=10.2 --channel https://mirrors.tuna.tsinghua.edu/anaconda/cloud/Paddle/
conda install -c nvidia cuda-nvcc_linux-64

2、安装nccl

安装nccl可以使框架实现单机多卡的多gpu使用。
我的另一篇文章有写,在服务器上,无网络安装的情况下,离线从源码安装nccl。这里写点在conda 环境中安装nccl方法。

conda install nccl -c conda-forge

3、验证

输入验证命令:

import paddle
paddle.utils.run_check()

正确安装,大概就是如下显示

更多推荐

paddle的安装