栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

NCCL源码安装

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

NCCL源码安装

文章目录
  • 1、下载
  • 2、解压
  • 3、填加环境变量

很多情况下,使用深度学习多GPU训练需要使用nccl,这里介绍最简单的待安装机器无法联网的安装方法,且安装包与系统版本无关,只与cuda版本有关。更多内容参见nccl 官网

1、下载

根据需要选择具体的架构,我们这里先x86,接着选与系统无关的那个:

下载好的文件是nccl_2.8.4-1+cuda11.2_x86_64.txz 这样一个以txz结尾的文件

2、解压
tar xvf nccl_2.8.4-1+cuda11.2_x86_64.txz

解压后,里边两个文件夹 include 和 lib

3、填加环境变量
vim ~/.bashrc

在最后加上:

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/where_your_path/nccl_2.8.4-1+cuda11.2_x86_64/include/:/where_your_path/nccl_2.8.4-1+cuda11.2_x86_64/lib

即可。
如果要安装给所有人用,可以将nccl_2.8.4-1+cuda11.2_x86_64文件夹拷到/usr/local下。
安装好以后,conda 环境,virtualenv等环境下都是可以用的。要注意的是,conda环境中有自己的cuda和cudnn,选择nccl对应版本时要按照conda环境中的cuda版本来进行处理
对于paddle可以输入:

import paddle
paddle.utils.run_check()

会返回所有可用gpu数量。

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/928540.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号