CUDA9.0环境mmdetection V0.5.7安装(踩坑)

本文详细介绍了在CUDA9.0环境下,如何安装mmdetection V0.5.7,特别指出必须使用pytorch 0.4.1版本,并提供了解决github上缺失文件问题的方法,包括从gitee克隆项目,切换到正确分支,创建虚拟环境,安装依赖库,编译及安装mmcv和mmdet的步骤。
摘要由CSDN通过智能技术生成

直接上干货,CUDA9.0版本对应的pytorch是0.4.1,使用其他版本pytorch不支持,一定要安装0.4.1版本,github上的mmdetection安装对应CUDA9.0的在branch0.4.1里,但是注意!!!!!!!!!!github上不知道为什么,pytorch0.4.1分支里的少一些文件我安装了好几次进到提示少deform_conv_cuda等文件,又对比主分支才发现少了很多文件.

解决办法:

1.git项目(不在github上)

git clone https://gitee.com/mirrors/mmdetection.git

2.切换到pytorch-0.4.1分支(针对CUDA9.0用户)

cd mmdetection

git checkout pytorch-0.4.1

3.安装

3.1创建虚拟环境(需要python3.5+)

conda create -n python3.5 python=3.5.4

conda activate python3.5

3.2安装依赖库(两种安装方式二选一)

 conda install pytorch=0.4.1 -c pytorch #pip install pytorch=0.4.1 -c pytorch

conda install cython #pip install cython

cd mmdetection#如果已经在此目录不需要此条命令

./compile.sh

3.3安装mmcv

git clone https://github.com/open-mmlab/mmcv.git
cd mmcv
pip install .  #不要少了 "."

4.安装mmdet

cd

评论 9
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值