| # Intel GPU Setup Guide for DeOldify |
|
|
| This guide covers how to set up DeOldify with Intel GPUs (Arc A-Series, Data Center GPU Max Series) using Intel Extension for PyTorch (IPEX). |
|
|
| ## Prerequisites |
|
|
| - **Intel GPU Driver**: Latest stable driver for your hardware. |
| - **Anaconda** or **Miniconda** installed. |
| - **Git** installed. |
| - **OS**: Windows 10/11 or Linux (Ubuntu 22.04 recommended). |
|
|
| ## Installation Steps |
|
|
| 1. **Clone the Repository** |
| ```bash |
| git clone https://github.com/thookham/DeOldify.git |
| cd DeOldify |
| ``` |
| |
| 2. **Create Intel Conda Environment** |
| We use a dedicated environment file that installs PyTorch with XPU support and Intel Extension for PyTorch. |
| ```bash |
| conda env create -f environment_intel.yml |
| ``` |
| |
| 3. **Activate Environment** |
| ```bash |
| conda activate deoldify_intel |
| ``` |
| |
| 4. **Download Weights** |
| Download the pretrained weights and place them in the `models/` directory: |
| - [ColorizeArtistic_gen.pth](https://github.com/thookham/DeOldify/releases/download/v2.0-models/ColorizeArtistic_gen.pth) |
| - [ColorizeStable_gen.pth](https://github.com/thookham/DeOldify/releases/download/v2.0-models/ColorizeStable_gen.pth) |
| - [ColorizeVideo_gen.pth](https://github.com/thookham/DeOldify/releases/download/v2.0-models/ColorizeVideo_gen.pth) |
| |
| ## Verification |
|
|
| Run the verification script to ensure your Intel GPU is detected: |
|
|
| ```bash |
| python verify_refactor.py |
| ``` |
|
|
| If successful, you should see: |
| - `Imports successful` |
| - `Device detection passed` |
| - `Device Name: Intel(R) Arc(TM)...` (or similar) |
|
|
| ## Troubleshooting |
|
|
| ### "XPU not available" |
| - Ensure you have installed the correct drivers. |
| - Verify that `intel-extension-for-pytorch` is installed: `pip list | grep intel` |
| - On Windows, ensure you are using the correct oneAPI components if manually installed. |
|
|
| ### Performance |
| - First run might be slower due to JIT compilation. |
| - Ensure you are using the `deoldify_intel` environment, not the standard one. |
|
|