Caffe2與PyTorch合并到同一個(gè)目錄下后父款,PyTorch推出了目前最新的穩(wěn)定版本0.4.1厢塘,但是通過(guò)以往在Anaconda環(huán)境中直接pip install torch torchvision的方式安裝PyTorch時(shí),安裝后import會(huì)報(bào)如下錯(cuò)誤:
libshm.so: undefined symbol: _ZTI24THRefcountedMapAllocator
搜索后得知該錯(cuò)誤是由于合并后的Caffe2和pre-build的PyTorch-0.4.1沖突所致,故考慮通過(guò)源碼編譯安裝PyTorch。
由于我同時(shí)需要使用Detectron,該工具包只支持Python2告抄。之前已經(jīng)按照Caffe2的官方指南將其安裝在系統(tǒng)的環(huán)境中,即/usr/local/python2.7/dist-packages
目錄下嵌牺,(執(zhí)行sudo python2 setup.py install
)打洼,使用系統(tǒng)自帶的Python2.7可以正常運(yùn)行Detectron。
但是這種安裝方式只能在Python2下使用PyTorch逆粹,我又同時(shí)需要在Python3下使用PyTorch募疮,如果在系統(tǒng)的Python3中再編譯安裝PyTorch就比較麻煩,不想這么干僻弹,所以考慮在Anaconda中新建一個(gè)環(huán)境阿浓,按照PyTorch GitHub中的源碼編譯安裝指南,安裝各種依賴蹋绽,在其他目錄下再clone一份PyTorch代碼芭毙,然后執(zhí)行python setup.py install
進(jìn)行編譯安裝,此時(shí)不需要sudo
就可以順利將PyTorch安裝至Anaconda的Python目錄下卸耘,即/home/huitr/anaconda3/envs/mmdetection/lib/python3.6/site-packages
退敦。此時(shí)安裝的PyTorch版本顯示為1.0.0a0+952df2b
,不知道能不能運(yùn)行mmdetection的代碼蚣抗,以后再試試吧侈百。