服务器(多GPU)caffe安装和编译
来源:互联网 发布:卖毛线比较好的淘宝店 编辑:程序博客网 时间:2024/05/17 02:33
一. 前提
多GPU交互在神经网络是常见的,所以在安装caffe之前需要安装NCCL,来保证多GPU之间的相互交流。
多GPU,这里指的是2个及2个以上英伟达显卡,而不是笔记本中的集显和独显。
二.安装NCCL
1.下载编译
shell终端
cd ncclmake CUDA_HOME=/user/local/cuda-7.5 test #注意自己的cuda路径
2.测试和配置环境变量
shell终端
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:./build/lib./build/test/single/all_reduce_test./build/test/single/all_reduce_test 10000000make install
注:make install 是自己添加,而官方原文没有。之所以这么加是因为在caffe 执行 cmake时候,cmake无法找到
非deb安装软件的路径,所以添加make install 是为了能让cmake识别到路径。
三.安装caffe
1.安装所需依赖
shell终端
sudo apt-get install --no-install-recommends build-essential cmake git gfortran libatlas-base-dev libboost-all-dev libgflags-dev libgoogle-glog-dev libhdf5-serial-dev libleveldb-dev liblmdb-dev libopencv-dev libprotobuf-dev libsnappy-dev protobuf-compiler python-all-dev python-dev python-h5py python-matplotlib python-numpy python-opencv python-pil python-pip python-protobuf python-scipy python-skimage python-sklearn
2.下载caffe
shell终端,cd到用户根目录
git clone https://github.com/NVIDIA/caffe.git caffe
3.编译caffe
shell终端
cp Makefile.config.example Makefile.configgedit Makefile.config
打开文本后,作出如下修改
取消下面这些话的前面注释符号#
USE_CUDNN := 1 USE_NCCL := 1ANACONDA_HOME := $(HOME)/anaconda #这里我们使用Anaconda环境下的pythonPYTHON_INCLUDE := $(ANACONDA_HOME)/include \ $(ANACONDA_HOME)/include/python2.7 \ $(ANACONDA_HOME)/lib/python2.7/site-packages/numpy/core/include \PYTHON_LIB := $(ANACONDA_HOME)/lib
在下面这些语句中加上#注释符号
#PYTHON_INCLUDE := /usr/include/python2.7 \ # /usr/lib/python2.7/dist-packages/numpy/core/include#PYTHON_LIB := /usr/lib
保存后,退出,编译caffe
sudo pip install -r caffe/python/requirements.txtcd caffemkdir buildcd buildmake all -jmake install -jmake runtest -j
0 0
- 服务器(多GPU)caffe安装和编译
- 【神经网络与深度学习】【CUDA开发】服务器(多GPU)caffe安装和编译
- Caffe安装以及GPU服务器设置并行
- windows系统下caffe的安装和编译及配置matlab接口和gpu加速
- 安装Caffe-Master(GPU和CPU)
- 【Caffe安装】Caffe安装(无GPU)--wanglei
- 【Caffe安装】Caffe安装(有GPU)--wanglei
- 集群服务器环境下安装Caffe深度学习库(GPU)
- Ubuntu14.04 安装Caffe(GPU)
- Mac Caffe安装(无GPU)
- ubuntu安装caffe完整版(GPU)
- ubuntu安装caffe +gpu
- windows配置caffe及matlab/python接口编译和调用(cpu/gpu)
- Caffe--Ubuntu 14.04 64bit caffe安装(无gpu)
- 在服务器GPU73上编译安装Caffe
- Mac10.12+XCode编译caffe(含GPU加速)
- Caffe-5.0--(GPU完整流程-0)--编译配置
- Ubuntu安装Caffe(无GPU)
- 搜狐狐友与腾讯微信两款社交软件的比较
- LeetCOde412详细分析并附求解代码
- 临时车牌相关问题
- uva 11292 勇者斗恶龙
- 嵌入式开发叙述
- 服务器(多GPU)caffe安装和编译
- 【Unity&NGUI】背包系统物品的累加
- Consul 入门笔记
- Read File in Python
- DIGITS安装和配置
- 关于逐项作用函数的用法
- 新手看Mockplus2.3
- markdown改变字体颜色和大小
- LeetCOde231&326详细分析并附求解代码