docs: update README

Author: okunator

README.md

@@ -3,7 +3,7 @@
 <div align="center">
 
 [![Generic badge](https://img.shields.io/badge/License-MIT-<COLOR>.svg?style=for-the-badge)](https://github.com/okunator/cellseg_models.pytorch/blob/master/LICENSE)
-[![PyTorch - Version](https://img.shields.io/badge/PYTORCH-1.7+-red?style=for-the-badge&logo=pytorch)](https://pytorch.org/)
+[![PyTorch - Version](https://img.shields.io/badge/PYTORCH-1.8+-red?style=for-the-badge&logo=pytorch)](https://pytorch.org/)
 [![Python - Version](https://img.shields.io/badge/PYTHON-3.7+-red?style=for-the-badge&logo=python&logoColor=white)](https://www.python.org/)
 <br>
 [![Github Test](https://img.shields.io/github/workflow/status/okunator/cellseg_models.pytorch/Tests?label=Tests&logo=github&style=for-the-badge)](https://github.com/okunator/cellseg_models.pytorch/actions/workflows/tests.yml)
@@ -57,10 +57,18 @@ Contains multi-task encoder-decoder architectures (U-Net-like) along with dedica
 
 ## Installation
 
+**Basic installation**
+
 ```shell
 pip install cellseg-models-pytorch
 ```
 
+**To install extra dependencies (training utilities and datamodules for open-source datasets) use**
+
+```shell
+pip install cellseg-models-pytorch[all]
+```
+
 ## Code Examples
 
 **Define Cellpose for cell segmentation.**