安装
- 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的安装
发布评论