當(dāng)前位置：
首頁(yè) >
ubuntu下载安装MaskRCNN-benchmark
發(fā)布時(shí)間：2023/11/27
47
豆豆
生活随笔
收集整理的這篇文章主要介紹了
ubuntu下载安装MaskRCNN-benchmark
小編覺(jué)得挺不錯(cuò)的,現(xiàn)在分享給大家,幫大家做個(gè)參考.
在window下，配個(gè)環(huán)境，一堆錯(cuò)誤，一周多都沒(méi)解決。換到ubuntu下，不到一天就配好了= =
強(qiáng)烈建議直接去ubuntu下配置，千萬(wàn)別在window環(huán)境下配置。
一、下載anaconda，創(chuàng)建虛擬環(huán)境
下載anaconda的步驟讀者可以去百度，很簡(jiǎn)單。隨后創(chuàng)建虛擬環(huán)境
conda create -n pytorch python=3.6.0
&emps;
二、下載CUDA與CUDNN
首先要更新驅(qū)動(dòng)，隨后進(jìn)行安裝。安裝步驟可參考博主之前的博客。
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
- 筆者的環(huán)境信息
CUDA：10.0
CUDNN:7.6.5
pytorch：1.2.0
- 去pytorch官網(wǎng)下載相應(yīng)的安裝包
https://pytorch.org/get-started/previous-versions/
- 安裝cocoapi
安裝coco數(shù)據(jù)集有關(guān)的api
進(jìn)入自己的目錄下，進(jìn)入虛擬環(huán)境
source activate pytorch
git clone https://github.com/cocodataset/cocoapi.git
cd cocoapi/PythonAPI
python setup.py build_ext install
- 安裝apex
這一步是最容易錯(cuò)的，因?yàn)楣倬W(wǎng)上下的版本很可能與自己的電腦的GCC不匹配，需要進(jìn)行回退
apex是由nvidia維護(hù)的一個(gè)支持混合精度分布式訓(xùn)練的第三方pytorch擴(kuò)展
這里不加sudo的話，很可能無(wú)法下載
sudo git clone https://github.com/NVIDIA/apex.git
降低apex的版本
cd apex
sudo git checkout f3a960f80244cf9e80558ab30f7f7e8cbf03c0a0
改變apex文件夾的權(quán)限
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
運(yùn)行demo
進(jìn)入到demo文件夾下
運(yùn)行
python3 webcam.py --min-image-size 300 MODEL.DEVICE cuda
運(yùn)行效果如下所示：
即使有GPU，運(yùn)行速度也很慢，一幀基本要0.6s
總結(jié)
以上是生活随笔為你收集整理的ubuntu下载安装MaskRCNN-benchmark的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。
- 上一篇： 腰椎间盘突出手术费用
- 下一篇： ubuntu下使用conda出现solv