ubuntu18.04 安装与卸载NCCL conda环境安装PaddlePaddle
cuda版本11.2
说明PaddlePaddle需要安装NCCL
登录官网
找到对应版本
官方提供了多种安装方式,本文使用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)!