ubuntu下载安装MaskRCNN-benchmark
生活随笔
收集整理的這篇文章主要介紹了
ubuntu下载安装MaskRCNN-benchmark
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
在window下，配個環境，一堆錯誤，一周多都沒解決。換到ubuntu下，不到一天就配好了= =
強烈建議直接去ubuntu下配置，千萬別在window環境下配置。
一、下載anaconda，創建虛擬環境
下載anaconda的步驟讀者可以去百度，很簡單。隨后創建虛擬環境
conda create -n pytorch python=3.6.0
&emps;
二、下載CUDA與CUDNN
首先要更新驅動，隨后進行安裝。安裝步驟可參考博主之前的博客。
https://blog.csdn.net/weixin_43828675/article/details/115708033?spm=1001.2014.3001.5501
https://blog.csdn.net/weixin_43828675/article/details/115710652?spm=1001.2014.3001.5501
?
三、安裝maskrcnn-benchmark
- 筆者的環境信息
CUDA：10.0
CUDNN:7.6.5
pytorch：1.2.0
- 去pytorch官網下載相應的安裝包
https://pytorch.org/get-started/previous-versions/
- 安裝cocoapi
安裝coco數據集有關的api
進入自己的目錄下，進入虛擬環境
source activate pytorch
git clone https://github.com/cocodataset/cocoapi.git
cd cocoapi/PythonAPI
python setup.py build_ext install
- 安裝apex
這一步是最容易錯的，因為官網上下的版本很可能與自己的電腦的GCC不匹配，需要進行回退
apex是由nvidia維護的一個支持混合精度分布式訓練的第三方pytorch擴展
這里不加sudo的話，很可能無法下載
sudo git clone https://github.com/NVIDIA/apex.git
降低apex的版本
cd apex
sudo git checkout f3a960f80244cf9e80558ab30f7f7e8cbf03c0a0
改變apex文件夾的權限
sudo chmod -R 777 ../apex
安裝
python setup.py install --cuda_ext --cpp_ext
- 安裝Mask-RCNN
cd ..
git clone https://github.com/facebookresearch/maskrcnn-benchmark.git
cd maskrcnn-benchmark
python setup.py build develop
運行demo
進入到demo文件夾下
運行
python3 webcam.py --min-image-size 300 MODEL.DEVICE cuda
運行效果如下所示：
即使有GPU，運行速度也很慢，一幀基本要0.6s
總結
以上是生活随笔為你收集整理的ubuntu下载安装MaskRCNN-benchmark的全部內容，希望文章能夠幫你解決所遇到的問題。
- 上一篇： 腰椎间盘突出手术费用
- 下一篇： ubuntu下使用conda出现solv