Spaces:
Runtime error
Runtime error
| ## 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 | |
| ``` |