facebook maskrcnn 安裝筆記

當前版本的maskrcnn嚴重依賴pythorch_nightly 1.0.0, 安裝過程中一不留神會由於各個包之間不兼容導致no module named maskrcnn_benchmark.

注意以下三點即可:

1. 先安裝ipython

2. 再單獨安裝torchvision。 conda install torchvision

3. 最後本地安裝pytorch-nightly. 先去https://anaconda.org/pytorch/pytorch-nightly/files下載合適的版本,包括python版本、cuda版本和cudnn版本。下載後使用 conda install --use-local xxxx安裝包名字   進行安裝

剩下步驟按照官方說明即可。

運行時如果報 no module named xxx 則直接搜索如何安裝xxx

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章