Common Guide for Running

Prepare

Intel GPU Driver

Refer to Install Intel GPU driver

Intel® oneAPI Base Toolkit

Refer to Install Intel® oneAPI Base Toolkit

Setup Running Environment

  1. Python

    We recommend you use Python 3.9 or newer. Check which version you’re running by using this command:

    python -V
    
  2. Create Virtual Environment

    We recommend you create a virtual Python environment called env_itex and install official Tensorflow* in it, like this:

    export ENV_NAME=env_itex
    rm -rf $ENV_NAME
    python -m venv $ENV_NAME
    source $ENV_NAME/bin/activate
    pip install --upgrade pip
    pip install tensorflow
    
  3. Install Intel® Extension for TensorFlow*

    pip install --upgrade intel-extension-for-tensorflow[gpu]
    

Running

Enable oneAPI Running Environment

source /opt/intel/oneapi/setvars.sh

Enable Virtual Running Environment

source env_itex/bin/activate

Run Script

Run your script with Intel® Extension for TensorFlow*.