您的位置:首页 > Web前端

深度学习_caffe-ubuntu-GPU 配置环境(0)

2016-01-11 18:45 507 查看
深度学习 -caffe .

昨天开始搭环境,其实害怕了很久不敢动手 后来突然想明白了,没事 没有什么事情是等到所有事情都完美了才动手的,一点点xue熊习(为什么我打不出来xue!!!!)。

根据是师兄给的文档,开始操练~

期间遇到的问题也都标注上了解决方法, 也不知道师兄从哪找的 我就大言不惭的写上了原创,如有侵权 请联系我 先抱歉一声 请收下我的歉意。

亲测可用。蠢笨的我都搞定了,这个文档还是挺清楚的!~

文档如下:

1 UBUNTU_CAFFE

sudo apt-get install -y g++ git freeglut3-dev

//遇到报错 尝试 sudo apt-get update

2安装依赖软件

sudo apt-get install -y libatlas-base-dev libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libboost-all-dev libhdf5-serial-dev libgflags-dev libgoogle-glog-dev liblmdb-dev protobuf-compiler

3配置 OpenCV 环境

cd ~/caffe

git clone https://github.com/jayrambhia/Install-OpenCV

cd Install-OpenCV/Ubuntu

sudo ./dependencies.sh

cd 2.4

sudo sh ./opencv2_4_10.sh

4安装 cudnn

cd ~/caffe

tar xvf cudnn-6.5-linux-R1.tgz

cd cudnn-6.5-linux-R1

sudo cp *.h /usr/local/include/.

将头文件拷贝到系统头文件目录下。

sudo cp lib* /usr/local/lib/.

将库文件拷贝到系统库文件目录下。

sudo ldconfig

//LLL添加 可能会出现这个报错:“/sbin/ldconfig.real: /usr/local/lib/libcudnn.so.6.5 不是符号连接”那是因为cuDNN没有链接成功,只能做一下硬链接。

$ sudo ln -sf /usr/local/lib/libcudnn.so.6.5.18 /usr/local/lib/libcudnn.so.6.5

//

5安装 caffe,将 caffe-master.zip 拷贝到工作目录下

cd ~/caffe

unzip caffe-master.zip

cd caffe-master

cp Makefile.config.example Makefile.config

vim Makefile.config

修改内容为: USE_CUDNN 行的注释去除,即 USE_CUDNN := 1

//删除#,然后 在命令模式中,连按两次大写字母Z,若当前编辑的文件曾被修改过,则Vi保存该文件后退出,返回到shell;若当前编辑的文件没被修改过,则Vi直接退出, 返回到shell。//

make all

6配置运行环境。

sudo vim /etc/ld.so.conf.d/caffe.conf

增加内容 /usr/local/cuda/lib64

sudo ldconfig

7运行

cd ~/caffe/caffe-master

sh data/mnist/get_mnist.sh

sh examples/mnist/create_mnist.sh

注释: vi examples/mnist/lenet_solver.prototxt

time sh examples/mnist/train_lenet.sh
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: