Ubuntu16.04安装Caffe2与PyTorch-1.0.0-beta

Caffe2与PyTorch合并到同一个目录下后,PyTorch推出了目前最新的稳定版本0.4.1,但是通过以往在Anaconda环境中直接pip install torch torchvision的方式安装PyTorch时,安装后import会报如下错误:

libshm.so: undefined symbol: _ZTI24THRefcountedMapAllocator

搜索后得知该错误是由于合并后的Caffe2和pre-build的PyTorch-0.4.1冲突所致,故考虑通过源码编译安装PyTorch。
由于我同时需要使用Detectron,该工具包只支持Python2。之前已经按照Caffe2的官方指南将其安装在系统的环境中,即/usr/local/python2.7/dist-packages目录下,(执行sudo python2 setup.py install),使用系统自带的Python2.7可以正常运行Detectron。

但是这种安装方式只能在Python2下使用PyTorch,我又同时需要在Python3下使用PyTorch,如果在系统的Python3中再编译安装PyTorch就比较麻烦,不想这么干,所以考虑在Anaconda中新建一个环境,按照PyTorch GitHub中的源码编译安装指南,安装各种依赖,在其他目录下再clone一份PyTorch代码,然后执行python setup.py install进行编译安装,此时不需要sudo就可以顺利将PyTorch安装至Anaconda的Python目录下,即/home/huitr/anaconda3/envs/mmdetection/lib/python3.6/site-packages。此时安装的PyTorch版本显示为1.0.0a0+952df2b,不知道能不能运行mmdetection的代码,以后再试试吧。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。