ubuntu16.04 caffe2(pytorch)安装

#-------------------------ZR----------------------------
#
#                利用annaconda安装caffe2
#
########!未经同意,不得转载!########

前面写的安装好 Detectron 后,本来一切正常,但是当我想要再次运行 py-faster-rcnn 之后,却意外的报错了。纳闷儿,完全没动过。错误点忘记记录了,只记得是关于 matplotlib 的。然后我就想是不是 caffe2 安装之后库出现冲突了,所以我就将 matplotlib 更新了一遍。果然,这下 py-faster-rcnn 正常了。

可是,高兴得太早了。回过头再运行 Detectron 的时候,这个又不行了,苍天啊!当时脑子一抽,干脆重新来过好了,所以我就照着自己之前实现的步骤重新安装 caffe2 。可是,奇怪的事情发生了,第一步 cmake .. 的时候就报错了:

Unknow Cmake Command: caffe2_interface_library

最关键的是,到最后也没解决,而且在我换一台电脑试出现了相同的问题,一气之下重装了系统,可是问题依旧一样。所以,我在想如果大家照着我的做会不会也是这样。。如果是的话,只能抱歉了,哈哈,总要走走坑大家才能更进步嘛。当然谁知道原因和解决了的话,可以告知一下,谢谢!

废话不多说了,最后我又仔细看了 caffe2 的官网安装,最后在选择利用 annaconda2 安装 caffe2 成功了(之前我的是没有  annaconda 的)。以下是步骤:

一、安装 annaconda2

这个简单,网上很多,以防万一,所有配置记录一下。我下载的是 Anaconda2-4.3.1-Linux-x86_64.sh 。(这个具体版本其实是不影响的)

二、下载pytorch

依赖项啊下载啊什么的前面有下载完成之后

cd pytorch

./scripts/build_anaconda.sh --install-locally --cuda 8.0 --cudnn 5

这个直接就会利用 conda build 自动编译 caffe2 了。中间会有一些可能需要你安装的东西,放心,会提示的。提示安装什么,就 conda install XXXX 就好了。

三、验证安装

# To check if Caffe2 build was successful
python2 -c 'from caffe2.python import core' 2>/dev/null && echo "Success" || echo "Failure"
 
# To check if Caffe2 GPU build was successful
# This must print a number > 0 in order to use Detectron
python2 -c 'from caffe2.python import workspace; print(workspace.NumCudaDevices())'

    正常结果如下:

zr@zr:~/Pytorch$ cd ~ && python -c 'from caffe2.python import core' 2>/dev/null && echo "Success" || echo "Failure"
Success
zr@zr:~$ python2 -c 'from caffe2.python import workspace; print(workspace.NumCudaDevices())'
1

大功告成!

你可能感兴趣的:(ubuntu16.04 caffe2(pytorch)安装)