安装mmpose+mmcv+mmdet

本人用的服务器版本是cuda11.4(实际安装的时候按照11.3版本)

 1.使用 conda 新建虚拟环境,并进入该虚拟环境

conda create -n openmmlab python=3.7 -y
conda activate openmmlab

这里的openmmlab是随便起的名字,当然你也可以随便叫

2.基于 PyTorch 官网安装 PyTorch 和 torchvision

conda install pytorch torchvision -c pytorch

这里我的服务器是cuda11.4,我找的是最接近的11.3进行适配的,因此我torch安装的版本是1.10.1

# CUDA 11.3
conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=11.3 -c pytorch -c conda-forge

同时我也建议根据上面的表格来找,然后去pytorch官网找就行

历史版本的记录在这里:Previous PyTorch Versions | PyTorch

3. 安装mmcv(我适用的conda下载的)

#使用pip
pip install mmcv-full==1.6.0 -f https://download.openmmlab.com/mmcv/dist/cu113/torch1.10.0/index.html

#使用conda
conda install mmcv-full=1.6.0 -y

cu后面写的是你cuda版本,同样torch版本也要对应(我这里就是11.3和1.10)

看了很多帖子都没有指定下载的版本号,我一开始也没有指定,自动下载的最新版1.6.2,但是出现了不兼容的问题,后来根据

conda search mmcv-full

可以查到兼容的版本,我这里最高支持到1.6.0

安装 MMCV — mmcv 1.7.1 文档这里也可以查到适合的版本

4. 下载mmdet

这里有对应的版本号(一定看好mmcv和mmdet的对应关系)

mmdetection/get_started.md at master · open-mmlab/mmdetection (github.com)

然后再通过

conda search mmdet

查找匹配目前版本的mmdet,我下载的是2.25.0版本,之前自动下载的2.25.2最新版也不兼容

mim install mmdet==2.25.0

基本的东西安装完了

下面进行测试

# 检查 Pytorch
import torch, torchvision
print('Pytorch 版本', torch.__version__)
print('torchvision版本', torchvision.__version__)
print('CUDA 是否可用',torch.cuda.is_available())

# 检查 mmcv
from mmcv.ops import get_compiling_cuda_version, get_compiler_version
print('CUDA版本', get_compiling_cuda_version())
print('编译器版本', get_compiler_version())

# 检查 mmtracking
import mmtracking

# 检查 mmpose
import mmpose
print('mmpose版本', mmpose.__version__)

我之前mmcv输出的就有问题,get_compiling_cuda_version()函数无法识别我的cuda,我用了research语句(上文搜索的那个命令),最终下载了1.6.0版本就可以成功识别了

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值