服务器安装nccl

xiaoxiao2022-07-06  37

问题:在服务器源码安装tensorflow-gpu时,需要配置nccl,但是服务器不联网,不能从github上下载,只能在本地安装

解决:

从官网下载nccl的安装包(对应自己的cuda版本)

下载地址:https://developer.nvidia.com/nccl/nccl-download

找对应的版本,服务器是centos系统,所以选择O/S agnostic版本

将下载好的安装包上传到服务器解压,之后的步骤和cudnn安装时完全相同(我的cuda安装在/home/myname/cuda目录下):

进入解压缩文件目录:

cp    include/nccl.h    /home/myname/cuda/include/

cp   /lib/libnccl*   /home/myname/cuda/lib64/

chmod a+r /home/myname/cuda/include/nccl.h   /home/myname/cuda/lib64/libnccl*

安装完毕

转载请注明原文地址: https://www.6miu.com/read-4968565.html

最新回复(0)