mirrors
/
mmpretrain
mirror of https://github.com/open-mmlab/mmpretrain.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[Docs] Fix `mmcv-full` in docs.

pull/1083/head
mzr1996 2022-10-09 08:14:06 +08:00
parent b9bb21738b
commit bf9f3bbdda
3 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docker/Dockerfile
View File

@ -14,7 +14,7 @@ RUN apt-get update && apt-get install -y ffmpeg libsm6 libxext6 git ninja-build
# Install MMCV
RUN pip install openmim
RUN mim install mmengine "mmcv-full>=2.0rc0"
RUN mim install mmengine "mmcv>=2.0rc0"
# Install MMClassification
RUN conda clean --all

2
docker/serve/Dockerfile
View File

@ -27,7 +27,7 @@ RUN pip install torchserve torch-model-archiver
ARG PYTORCH
ARG CUDA
RUN pip install mmengine
RUN ["/bin/bash", "-c", "pip install mmcv-full==${MMCV} -f https://download.openmmlab.com/mmcv/dist/cu${CUDA//./}/torch${PYTORCH}/index.html"]
RUN ["/bin/bash", "-c", "pip install mmcv==${MMCV} -f https://download.openmmlab.com/mmcv/dist/cu${CUDA//./}/torch${PYTORCH}/index.html"]
RUN pip install mmcls==${MMCLS}
RUN useradd -m model-server \

8
docs/zh_CN/get_started.md
View File

@ -47,7 +47,7 @@ conda install pytorch torchvision cpuonly -c pytorch
```shell
pip install -U openmim
mim install mmengine "mmcv-full>=2.0rc0"
mim install mmengine "mmcv>=2.0rc0"
```
**第 2 步** 安装 MMClassification
@ -145,10 +145,10 @@ MMCV 包含 C++ 和 CUDA 扩展，因此其对 PyTorch 的依赖比较复杂。M
要使用 pip 而不是 MIM 来安装 MMCV，请遵照 [MMCV 安装指南](https://mmcv.readthedocs.io/zh_CN/latest/get_started/installation.html)。
它需要你用指定 url 的形式手动指定对应的 PyTorch 和 CUDA 版本。
举个例子，如下命令将会安装基于 PyTorch 1.10.x 和 CUDA 11.3 编译的 mmcv-full。
举个例子，如下命令将会安装基于 PyTorch 1.10.x 和 CUDA 11.3 编译的 mmcv。
```shell
pip install "mmcv-full>=2.0rc0" -f https://download.openmmlab.com/mmcv/dist/cu113/torch1.10/index.html
pip install "mmcv>=2.0rc0" -f https://download.openmmlab.com/mmcv/dist/cu113/torch1.10/index.html
```
### 在 CPU 环境中安装
@ -166,7 +166,7 @@ MMClassification 中几乎所有的模型都不会依赖这些算子。
```shell
!pip3 install openmim
!mim install mmengine "mmcv-full>=2.0rc0"
!mim install mmengine "mmcv>=2.0rc0"
```
**第 2 步** 从源码安装 MMClassification