win10_ubuntu1604_caffe安装

来源:互联网 发布:怎么练淘宝模特动作 编辑:程序博客网 时间:2024/06/05 00:05

1. 安装双系统

本人电脑是(惠普(HP)暗影精灵II代Pro 精灵绿 15.6英寸游戏笔记本(i7-7700HQ 8G 128GSSD+1T GTX1050Ti 4G独显 IPS FHD), 电脑默认安装win10在ssd上,因为需要跑深度学习所以安装Ubuntu16.04.(14.04出错)

(1)用ultraiso制作U盘启动盘,然后进行安装。
安装比较简单,就是需要注意的是 以前如果非UEFI启动的机器需要分区/boot,现在只需要新建uefi文件格式分区。然后安装向导选择新建的这个区,需要关闭security boot。

(2)安装完成之后,此时开机无法进入ubuntu,这时需要进入bios中选择启动项,将ubuntu引导顺序放在windows之前,因为win10无法引导ubuntu。这样就完成系统安卓。

2.安装nvidia驱动

因为nouveau驱动的影响,ubuntu安装后无法登入桌面,所以在ubuntu系统启动显示登录界面后,需要按ctrl+alt+F1进入tty文本模式进入下面的操作
sudo vim /etc/modprobe.d/blacklist.conf
添加

blacklist vga16fb blacklist nouveau blacklist rivafb blacklist rivatv blacklist nvidiafb

更新内核
sudo update-initramfs -u

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
寻找合适的驱动版本
ubuntu-drivers devices

sudo service lightdm stop
安装完成后重启
sudo apt-get install nvidia-381
sudo reboot
重启系统后,执行下面的命令查看驱动的安装状态显示安装成功
sudo nvidia-smi

3. 安装cuda

安装caffe需要安装cuda库
在官网下载cuda,使用迅雷快速下载,然后拷贝到ubuntu下,安装命令如下:

sudo dpkg -i cuda-repo-ubuntu1404-8-0-local_8.0.44-1_amd64.deb
sudo apt-get update
sudo apt-get install cuda
然后cd /usr/local/cuda/sample中编译例子,运行deviceQuery,不重启电脑运行失败,然后重启电脑,安装成功,如果要查看显卡的内存使用情况,可以用nvidia-smi查看。

4. 安装cudnn

https://developer.nvidia.com/cudnn下载
sudo cp lib* /usr/local/cuda/lib64/
sudo cp cudnn.h /usr/local/cuda/include/
更新软连接
cd /usr/local/cuda/lib64/sudo rm -rf libcudnn.so libcudnn.so.6 #删除原有动态文件

sudo ln -s libcudnn.so.6.0.21 libcudnn.so.6 #生成软衔接 sudo ln -s libcudnn.so.6 libcudnn.so #生成软链接

5.安装caffe

我是直接参考阅读官网的安装教程,没有遇到什么问题。
最后执行 make all -j4。
然后编译Python接口需要安装
make pycaffe,出现找不到numpy这些头文件,所以安装一下numpy,scipy,
matplotlib。命令如下:
sudo apt-get install python-numpy python-scipy python-matplotlib

4.理论与实战

运行cifar10例子,其实比较简单,该cifar10_quick_train.prototxt网络,由卷积层,POOLing层,全连接层组成,网络比较简单,测试准确率为76%。

学习python接口进行学习我主要参考python jupyter这个教程,可查看每一层的参数。

如何可视化网络,学习了使用netscope。

理论方面,自学斯坦福ufdl教程,理解了反向传播的原理。

实践方面,看到caffe做图像分类时,都需要减去所有训练样本的均值,工具在caffe中,这个答案需要参考斯坦福ufdl教程的白化预处理部分,其他预处理还会有crop_size,mirror,scale,mean_value,force_gray。caffe源码分析可以参考这位大神的博客。当然为了搞清楚caffe原理也可以看看这个从零开始山寨caffe博客教程,这个会对caffe里面各种设置理解的更加清楚。比如在prototxt文件中,我们会看到 scale 为0.00390625,这个就是为了将图像输入值压缩到[0,1]之间。

计划: 接下来尝试去增减一些层数看效果如何。可以参考这个benchmark。

原创粉丝点击