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进行处理,非常感谢!