您现在的位置: 首页>> 资讯 >> 正文

caffe框架主要用来做啥_caffe框架|环球精选

来源: 互联网


(资料图)

1、Caffe需要预先安装一些依赖项,首先是CUDA驱动。

2、不论是CentOS还是Ubuntu都预装了开源的nouveau显卡驱动(SUSE没有这种问题),如果不禁用,则CUDA驱动不能正确安装。

3、以Ubuntu为例,介绍一下这里的处理方法,当然也有其他处理方法。

4、123456#sudovi/etc/modprobe.d/blacklist.conf#增加一行:blacklistnouveausudoapt-get--purgeremovexserver-xorg-video-nouveau#把官方驱动彻底卸载:sudoapt-get--purgeremovenvidia-*#清除之前安装的任何NVIDIA驱动sudoservicelightdmstop#进命令行,关闭XserversudokillallXorg然后下载并安装NVIDIA CUDA驱动包,接着安装安装BLAS、OpenCV、Boost这三个库。

5、BLAS数学库可以是ATLAS, MKL, 或 OpenBLAS,OpenCV要求2.4以上版本,Boost要求1.55版本以上。

6、可选安装Python MATLAB Caffe 库,还有numpy ,pandas 之类的Python类库。

7、安装MATLAB,以确保mex 在path路径中,这些就够你折腾一个星期了,完事以后就可以编译安装了:12345cpMakefile.config.exampleMakefile.config#AdjustMakefile.config(forexample,ifusingAnacondaPython,orifcuDNNisdesired)makeallmaketestmakeruntest哦对了,Caffe 显卡要求:Titan Xs, K80s, GTX 980s, K40s, K20s, Titans, and GTX 770s。

本文到此分享完毕,希望对大家有所帮助。

相关新闻