| # Linux | |
| Please follow the primary README.md of this repo. | |
| # Windows | |
| Windows users may stumble when installing the package `triton`. | |
| You can choose to run on **CPU** without `xformers` and `triton` installed. | |
| To use **CUDA**, please refer to [issue#24](https://github.com/XPixelGroup/DiffBIR/issues/24) to try solve the problem of `triton` installation. | |
| # MacOS | |
| <!-- Currenly only CPU device is supported to run DiffBIR on Apple Silicon since most GPU acceleration packages are compatible with CUDA only. | |
| We are still trying to support MPS device. Stay tuned for our progress! --> | |
| You can try to set up according to the following steps to use CPU or MPS device. | |
| 1. Install **torch (Preview/Nighly version)**. | |
| ```bash | |
| # MPS acceleration is available on MacOS 12.3+ | |
| pip install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cpu | |
| ``` | |
| Check more details in [official document](https://pytorch.org/get-started/locally/). | |
| 2. Package `triton` and `xformers` is not needed since they work with CUDA. Remove the related packages. | |
| Your requirements.txt should look like: | |
| ```bash | |
| # requirements.txt | |
| pytorch_lightning==1.4.2 | |
| einops | |
| open-clip-torch | |
| omegaconf | |
| torchmetrics==0.6.0 | |
| opencv-python-headless | |
| scipy | |
| matplotlib | |
| lpips | |
| gradio | |
| chardet | |
| transformers | |
| facexlib | |
| ``` | |
| ```bash | |
| pip install -r requirements.txt | |
| ``` | |
| 3. [Run the inference script](https://github.com/XPixelGroup/DiffBIR#general_image_inference) and specify `--device cpu` or `--device mps`. Using MPS can accelarate your inference. | |
| You can specify `--tiled` and related arguments to avoid OOM. |