Spaces:
Runtime error
Runtime error
File size: 1,425 Bytes
f5fdf51 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 |
## Installation Guide
YOLO-World is built based on `pytorch=1.11.0` and `mmcv=2.0.0`.
We provide the `requirements` files in [./requirements](./../requirements/):
* `basic_requirements`: training, finetuning, evaluation.
* `demo_requirements`: running YOLO-World [demos](./../demo/).
* `onnx_requirements`: converting YOLO-World to ONNX or TFLite models (TFLite is coming soon).
#### Install `MMCV`
YOLO-World adopts `mmcv>=2.0.0`. There are several ways to install `mmcv`
**1. using `openmim`**:
see more in [official guide](https://github.com/open-mmlab/mmcv/tree/master?tab=readme-ov-file#install-mmcv-full).
```bash
pip install openmim
mim install mmcv==2.0.0
```
**2. using `pip`**:
go to [install-with-pip](https://mmcv.readthedocs.io/en/latest/get_started/installation.html#install-with-pip) to select the pip index.
```bash
# cuda=11.3, torch=1.11
pip install mmcv==2.0.0 -f https://download.openmmlab.com/mmcv/dist/cu113/torch1.11/index.html
# cuda=11.7, torch=1.13
pip install mmcv==2.2.0 -f https://download.openmmlab.com/mmcv/dist/cu117/torch1.13/index.html
# cuda=12.1, torch=2.1
pip install mmcv==2.1.0 -f https://download.openmmlab.com/mmcv/dist/cu121/torch2.1/index.html
```
**3. using `whl`**
go to [index packages](https://download.openmmlab.com/mmcv/dist/cu117/torch1.13/index.html) to find a suitable version and download.
```bash
pip install mmcv-2.0.1-cp38-cp38-manylinux1_x86_64.whl
``` |