Ubuntu安装pytorch和mmdetection
mmdetection不推荐1.11.0版本的torch**pytorch官方中torch历史版本:[https://pytorch.org/get-started/previous-versions/](https://pytorch.org/get-started/previous-versions/)```pythonconda create -n mmdet python=3.7
·
1.首先新建conda虚拟环境,安装pytorch
mmdetection不推荐1.11.0版本的torch
pytorch官方中torch历史版本:https://pytorch.org/get-started/previous-versions/
conda create -n mmdet python=3.7
查看conda环境
conda info --env
安装pytorch
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge
2.安装mmcv-full
mmdetection中文文档:
https://mmdetection.readthedocs.io/zh_CN
pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/{cu_version}/{torch_version}/index.html
这里的cu_version和torch_version是根据自己版本而定的,不清楚的可以看下面这个网址:
https://gitee.com/cubone/mmcv
根据上面安装的pytorch版本,我的mmcv-full安装指令如下:
pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.8.0/index.html
3.下载、编译、安装mmdetection
首先,安装git
conda install git
安装mmdetection
git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
pip install -r requirements/build.txt
pip install -v -e . # or "python setup.py develop"
若服务器不能翻墙,就从本机上下载后传到服务器即可
4.测试
python tools/train.py -h
更多推荐
已为社区贡献1条内容
所有评论(0)