caffe缺少NCCL库导致不能多GPU训练问题(改makefile版)

最近用caffe多GPU联调时遇到问题,如果正常用caffe的多gpu训练而不加初始化模型,末尾加--gpu=0,1,不会有问题;然而当用pretrained模型finetune网络时,如--weights=xxx.caffemodel,则会报如下错误:
Multi-GPU execution not available - rebuild with USE_NCCL
提示nccl没有安装,于是百度了一下,需要下载nccl源码安装,然后就是修改caffe的编译选项。由于我用的make编译的,需要该2个地方:
1.在Makefile.config文件中加

USE_NCCL := 1
INCLUDE_DIRS += /home/xd133/nccl/build/include
LIBRARY_DIRS += /home/xd133/nccl/build/lib

2.在Makefile中加

ife
  • 0
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
### 回答1: 在安装 Apt Caffe GPU 本时,其安装目录会根据不同的操作系统和安装方式有所差异。 在 Linux 系统上,通常会将 Apt Caffe GPU 本安装在默认的 /usr/local 目录下。具体安装目录为 /usr/local/opt/caffe-gpu。 在 macOS 系统上,Apt Caffe GPU 本的安装目录也常为 /usr/local 目录下。具体安装目录为 /usr/local/opt/caffe-gpu。 需要注意的是,以上的目录仅为参考,实际的安装目录可能会因为个人设置、操作系统本等因素而有所不同。可以通过查看安装过程中的输出信息或者参考相应的文档来确定确切的安装目录。 另外,在安装完 Apt Caffe GPU 本后,我们也可以通过配置相关环境变量来使用相应的和工具。比如,可以设置 LD_LIBRARY_PATH 环境变量指向 Apt Caffe GPU 的安装目录,以便系统能够正确查找和加载相关的动态文件。 总之,Apt Caffe GPU 本的安装目录可能会根据不同的系统和安装方式有所变化,但一般来说会位于 /usr/local 目录下。可以通过查看安装过程中的输出信息或者相关文档来确定确切的安装目录。 ### 回答2: apt caffeGPU本默认安装目录是在/usr/local/下。apt-get命令在安装软件时会将其默认安装到该目录下,因此apt caffeGPU本也不例外。 在安装完成之后,相关的文件和目录会被安装到/usr/local/caffe/目录下。其中包括caffe的源代码、编译后的可执行文件、配置文件、模型文件等。而GPU相关的文件和目录则会在/usr/local/cuda/下。 在这些目录中,我们可以找到caffe编译后的可执行文件caffe,以及其他与caffe相关的工具。此外,也可以在/usr/local/cuda/目录下找到与GPU相关的文件和头文件。 值得一提的是,以上路径仅仅是默认路径,用户在安装时也可以选择其他路径进行安装。在进行安装之前,可以通过apt-get命令的参数来指定安装目录,以达到自定义安装路径的目的。但需要注意的是,如果选择了非默认目录,在使用时需要注意相关文件和目录的路径。 总结起来,apt caffe GPU本的默认安装目录是/usr/local/caffe/,与之相关的GPU文件和目录则位于/usr/local/cuda/下。用户也可以选择其他自定义路径进行安装,只需在安装时进行相应的设置即可。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值