首页 > 代码库 > Ubuntu14.04安装Caffe(GPU)

Ubuntu14.04安装Caffe(GPU)

双显卡不适合安装caffe,因为通常集成显卡不能切断,而桌面的图像必须有集成显卡控制,导致NVIDIA显卡驱动安装后桌面黑屏

不同显卡驱动对应不同的CUDA,例如:GTM550-CUDA6.5.14;K5000-CUDA7.0.28

装CUDA似乎可以不用装NVIDIA,因为CUDA自带驱动,另外,可以不装NVIDIA-Toolkit,不用管nvcc的事

安装完Ubuntu后关闭自动更新,若其自动更新了显卡驱动,可能会与caffe不兼容

安装过程,因为平台不同,设备不同,操作者不同,会遇到各种奇怪的问题,与报错信息

 

caffe的优势:

1.上手快:模型与相应优化均以文本形式而非代码形式给出,caffe给出了模型的定义,最优化设置以及预训练的权重

2.速度快:与CuDNN结合使用,测试AlexNet模型,在K40上处理每张图只需1.17ms

3.模块化:方便扩展到新任务,设置上可以使用caffe提供的各层类型来定义自己的模型

4.开发性:公开的代码与参考模型用于再现

5.社区好:可以通过BSD-2参于开发与讨论。

Ubuntu14.04安装Caffe(GPU)