ubuntu 18.04.1安装配置caffe2

1.ubuntu下安装一些框架、软件啥的真是处处是坑,在ubuntu 16.04和18.04.1上按照官 给的教程安装caffe2在最后一步(sudo python setup.py install)出错了,瞎折腾了一通之后终于搞定了,在此记录一下安装过程。
2.安装过程
首先参考官方教程,安装一些依赖库:

接下来这一步是一个坑,官 给出的是:

如果敲入这一命令,会出现如下错误(只针对本电脑,其他电脑可能会很顺利地成功安装):

Notices:
在运行python setup.py install会自动下载一个压缩文件mklml_lnx_2019.0.3.20190220.tgz(这个下载速度很慢,60多MB要40分钟左右mklml_lnx_2019.0.3.20190220.tgz)到pytorch/third_party/ideep/mkl-dnn/external目录下并自动解压,所以,为了保险起见,在运行下面的make命令之前先执行一下python setup.py install。

百思不得其解,实在没办法,我换一种方式搞:

Notices:
若要使caffe2支持GPU,在执行下面的make和make install命令时应先正确安装相应版本的CUDA和cudnn,否则安装后只支持cpu模式,接下去的make和make install花的时间有点长,耐心等待。

(1)进入git clone下载下来的pytorch目录,执行命令:

执行后生成一个build文件夹

(2)cd进入build目录,执行:

ok,运气不错,没出现错误。

(3)测试一下:

输出 ‘Failure’,心又凉了,换一种可以输出详细错误信息的测试方式:
进入~目录,运行python,然后import一下:

出现如下错误提示

尝试解决方案:

再测试一下:

(4)正确安装了CUDA和cudnn之后测试:

发现提示说caffe2不支持GPU模式,只支持cpu模式,解决办法:

将Caffe2添加到Python 库路径中:

末尾添加:

source一下使之生效:

ok,搞定。

声明:本站部分文章及图片源自用户投稿,如本站任何资料有侵权请您尽早请联系jinwei@zod.com.cn进行处理,非常感谢!

上一篇 2019年3月11日
下一篇 2019年3月11日

相关推荐