cuda10安装_Mmdetection的安装和使用

Mmdetection的安装和使用

环境：ubuntu16.04+cuda10.1+gcc5.5.0+pytorch1.3.1

参考网址：https://github.com/open-mmlab/mmdetection

第一步：安装环境和相关依赖库

按照教程中的installation：

conda create -n open-mmlab python=3.7
conda activate open-mmlab
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
pip install mmcv
python setup.py develop  # or "pip install -v -e ."

第二步：准备数据集（VOC2007）

我用的是VOC2007数据集，主要是为了测试一下环境是否安装成功，是否可以跑通代码

mkdir data

将下载的VOC2007数据集放在data目录下即可，如下所示：

mmdetection
├── mmdet
├── tools
├── configs
├── data
│   ├── VOCdevkit
│   │   ├── VOC2007

第三步：运行train.py：

python tools/train.py configs/pascal_voc/faster_rcnn_r50_fpn_1x_voc0712.py

报错，因为datasets中加载VOC的时候是VOC2007和VOC2012同时输入，我只下载了VOC2007,所以先把涉及到VOC2012的代码注释configs/pascal_voc/faster_rcnn_r50_fpn_1x_voc0712.py中修改如下:

之后再次运行train.py python tools/train.py configs/pascal_voc/faster_rcnn_r50_fpn_1x_voc0712.py

报错如下：

我google了一下，是环境配置问题，我自己的服务器中有cuda9.0,cuda10.0,cuda10.1三个版本，默认的是cuda10.0版本，我刚开始下载的pytorch1.3.1是cuda10.1版本，所以出现了不匹配的问题，所以我的想法是把cuda10.1路径添加到系统环境变量中去命令如下：

vim ~/.bashrc
export CUDA_HOME=/usr/local/cuda-10.1
export PATH="/usr/local/cuda-10.1/bin:$PATH"
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda-10.1/lib64"
export LIBRARY_PATH="$LIBRARY_PATH:/usr/local/cuda-10.1/lib64"
source ~/.bashrc

之后进行重新编译，一定要删掉mmdetection/build文件夹，之后再次编译，我之后的编译是按照下面命令来的：

python setup.py build_ext --inplace
export PYTHONPATH=${PWD}:$PYTHONPATH
pip install -r requirements.txt

重新编译完成之后，再次运行python tools/train.py configs/pascal_voc/faster_rcnn_r50_fpn_1x_voc0712.py

可以成功运行：

附注：

1.服务器之间互传文件命令： scp -P 1085 -r /workspace1/zigangzhao/TensorFlowBDD zigangzhao@222.20.79.224:/workspace/zigangzhao/

cuda10安装_Mmdetection的安装和使用相关推荐

win10基于anaconda下的tensorflow2.0.0及cuda10.0、cudnn安装成功
anaconda下安装tensorflow2.0.0 cuda10.0下载安装安装cudnn 安装tensorflow2.0.0 cuda10.0下载安装 CUDA Toolkit 10.0 Dow ...
WIN10+CUDA10.1+cudnn+Anaconda安装TensorflowGPU版本
在WIN10下基于CUDA10.1和CUDNN10.0在Anaconda Python3.7下安装TensorflowGPU版本 ====================== 更新2019/8/3 = ...
Win10+Cuda10.2的Pytorch安装
Win10+Cuda10.2的Pytorch安装今天是2022年10月底,最新版本Cuda稳定版已不支持Cuda10.2 由于电脑是非安培架构(30系)的NV的GPU,故装了Cuda10.2 (如果 ...
ubuntu16.04下Nvidia T1000显卡驱动460、CUDA10.2、CUDNN8安装
准备工作架构与GPU型号:非官方介绍,页面搜索一下GPU就行.这个表显示我应该装CUDA10,但是先8.0凑和着.理论依据,要用9的话会出现这个问题 Pascal (CUDA 8 and later ...
Jeston nano配置（三）卸载cuda10.2安装cuda9 并安装tensorflow-gpu
(一)卸载cuda10.2 参考帖子: Nvidiajetsonnano配置CUDA9CUDNN7.1.5 - 百度文库 ourkix的博客:Nvidia jetson nano 配置 CUDA9 C ...
Win10 + CUDA10.1+pytorch手动安装
1.安装环境 OS: Win10 专业 x64 Python: Python 3.7.7(通过Anaconda软件自带安装) Anaconda: Anaconda3 ...
gpu tensorflow 怎么看是否安装成功_安装keras模块以及使用GPU加速
在神经网络的实现中,需要使用GPU进行加速,并使用keras模块来构建神经网络(当然,也可以使用pytorch).但是,安装并配置这些东西真的是比编程还难,过程中遇到各种问题,然后去搜解决方法,但都五 ...
Ubuntu16.04显卡驱动安装和Cuda安装
环境: 操作系统:Ubuntu 16.04 x64,USB 安装独显:Geforce GTX 1050,本地安装 Cuda 10.0 网络在线安装一.安装Geforce GTX 1050 驱动安 ...
【服务器管理】Ubuntu18.04下安装TensorRT(已经安装CUDA和cudnn的情况下)
本人实践经历在运行深度学习项目,常常遇到警告内容: 021-06-16 15:51:21.131229: W tensorflow/stream_executor/platform/default/ ...
python安装numba_Jetson Nano安装Numba遇到的一些坑
一开始简单粗暴的pip3 install numba,结果安装llvmlite时候报错. 然后百度,参考了CSDN上的一篇博客以下为博客中内容: 首先numba安装需要依赖包llvmlite,llv ...

cuda10安装_Mmdetection的安装和使用

cuda10安装_Mmdetection的安装和使用相关推荐

最新文章

热门文章