docs/intel_gpu_setup.md

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

   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.

   conda env create -f environment_intel.yml
   

3. Activate Environment

   conda activate deoldify_intel
   

4. Download Weights Download the pretrained weights and place them in the models/ directory:

   • ColorizeArtistic_gen.pth
   • ColorizeStable_gen.pth
   • ColorizeVideo_gen.pth

Verification

Run the verification script to ensure your Intel GPU is detected:

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.