Ubuntu下Caffe安装记录

亦凉 2022-06-01 03:29 178阅读 0赞

## 总体过程 ##

*  [Ubuntu16.04 Caffe 安装步骤记录（超详尽）][Ubuntu16.04 Caffe]
 *  手动安装Nvidia显卡驱动，安装Cuda时只需要安装ToolKit，也不要安装opengl

## 问题解决 ##

*  [修改 Ubuntu16.04 源为清华大学——解决依赖包无法安装问题][Ubuntu16.04]
 *  [解决APT-GET更新源报错：W: UNKNOWN MULTI-ARCH TYPE ‘NO’ FOR PACKAGE ‘COMPIZ-GNOME’][APT-GET_W_ UNKNOWN MULTI-ARCH TYPE _NO_ FOR PACKAGE _COMPIZ-GNOME]
 *  [ctrl+alt+F1～6进入不了字符界面，黑屏的解决办法][ctrl_alt_F1_6]（注意：添加nomodeset，不要删除quiet splash）
 *  [安装NVIDIA驱动][NVIDIA]（单独安装驱动，防止caffe make runtest报错，安装cuda时不用再安装驱动）
 *  [ cuda程序执行出错: libcudart.so.8.0: cannot open shared object file: No such file or directory][cuda_ libcudart.so.8.0_ cannot open shared object file_ No such file or directory]
 *  [ Ubuntu安装CUDA报错：/usr/lib/nvidia-375/libEGL.so.1 不是符号连接 ][Ubuntu_CUDA_usr_lib_nvidia-375_libEGL.so.1 _]
 *  [import caffe会出现ImportError: No module named google.protobuf.internal——sudo pip install protobuf ][import caffe_ImportError_ No module named google.protobuf.internal_sudo pip install protobuf]
 *  [关于关不了机的尝试][Link 1]:nvidia的显卡驱动可能是罪魁祸首，最后安装还是软件源推荐的版本，cuda里的不要安装。

## 命令行训练Mnist ##

# 进入caffe目录，下载mnist数据
    ./data/mnist/get_mnist.sh
    # 转换数据格式 （用sudo防止出现Cannot create Cublas handle. Cublas won't be available.错误）
    sudo ./examples/mnist/create_mnist.sh
    # 训练
     sudo ./examples/mnist/train_lenet.sh

## PyCharm训练Mnist ##

1.  需要用sudo执行pycharm，还是防止出现Cannot create Cublas handle. Cublas won’t be available.错误；
2.  需要将example/mnist/下的一些文件中定义的训练数据等路径配置改成绝对路径；
3.  代码：
    
        # coding=utf-8
        
        import caffe
        
        caffe.set_mode_gpu()
        solver = caffe.SGDSolver('/home/guopin/caffe/examples/mnist/lenet_solver.prototxt')
        solver.solve()

[Ubuntu16.04 Caffe]: http://blog.csdn.net/yhaolpz/article/details/71375762
[Ubuntu16.04]: http://blog.csdn.net/dearsq/article/details/51492847
[APT-GET_W_ UNKNOWN MULTI-ARCH TYPE _NO_ FOR PACKAGE _COMPIZ-GNOME]: http://blog.csdn.net/jk123vip/article/details/51078797
[ctrl_alt_F1_6]: https://www.cnblogs.com/shrimp-can/p/5134426.html
[NVIDIA]: http://blog.csdn.net/SEUlvjian/article/details/78050832
[cuda_ libcudart.so.8.0_ cannot open shared object file_ No such file or directory]: http://blog.csdn.net/u010454261/article/details/71268325?utm_source=itdadao&utm_medium=referral
[Ubuntu_CUDA_usr_lib_nvidia-375_libEGL.so.1 _]: http://blog.csdn.net/teavamc/article/details/78334707
[import caffe_ImportError_ No module named google.protobuf.internal_sudo pip install protobuf]: https://www.jianshu.com/p/1e405b9fe973
[Link 1]: http://www.52nlp.cn/%E6%B7%B1%E5%BA%A6%E5%AD%A6%E4%B9%A0%E4%B8%BB%E6%9C%BA%E7%8E%AF%E5%A2%83%E9%85%8D%E7%BD%AE-ubuntu-16-04-nvidia-gtx-1080-cuda-8