自学内容网 自学内容网

ubuntu18.04 安装与卸载NCCL conda环境安装PaddlePaddle

cuda版本11.2

说明PaddlePaddle需要安装NCCL

1、Log in | NVIDIA Developer

登录官网

找到对应版本

官方提供了多种安装方式,本文使用Local installers (x86)本地安装

点击对应的版本下载如:

nccl-local-repo-ubuntu1804-2.8.4-cuda11.2_1.0-1_amd64.deb

安装步骤:

sudo dpkg -i nccl-local-repo-ubuntu1804-2.8.4-cuda11.2_1.0-1_amd64.deb

# 更新源镜像

sudo apt update

# 安装 nccl

sudo apt install libnccl2=2.8.4-1+cuda11.2 libnccl-dev=2.8.4-1+cuda11.2

安装nccl的命令来源于:

配置环境变量:

nccl默认的安装目录是/usr/lib/x86_64-linux-gnu,在终端模拟器中使用vim ~/.bashrc 打开配置文件,按插入键 i ,在 ~/.bashrc 文件中添加以下内容之后保存文件:

#设置cuda库的目录
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
#将nccl添加到LD_LIBRARY_PATH中
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/lib/x86_64-linux-gnu

先按Esc退出,再输入:wq,最后执行source ~/.bashrc使文件的配置生效

2、卸载linux系统NCCL安装与卸载_linux_多想常做-Linux

3、安装PaddlePaddleLinux 下的 Conda 安装-使用文档-PaddlePaddle深度学习平台

# 创建conda虚拟环境
conda create -n paddle_env python=paddle

# 激活环境
conda activate paddle

# GPU 版的 PaddlePaddle
# 对于 CUDA 11.2,需要搭配 cuDNN 8.2.1(多卡环境下 NCCL>=2.7),安装命令为:

conda install paddlepaddle-gpu==2.6.1 cudatoolkit=11.2 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/Paddle/ -c conda-forge

# 测试安装是否成功

#进入 python 解释器,输入
import paddle
paddle.utils.run_check()

#如果出现PaddlePaddle is installed successfully!,说明您已成功安装。


原文地址:https://blog.csdn.net/m0_50907280/article/details/143607389

免责声明:本站文章内容转载自网络资源,如本站内容侵犯了原著者的合法权益,可联系本站删除。更多内容请关注自学内容网(zxcms.com)!