Common Guide for Running

Prepare

Intel GPU Driver

Refer to Install Intel GPU dirver

Intel® oneAPI Base Toolkit

Refer to Install Intel® oneAPI Base Toolkit

Setup Running Environment

  1. Python

    This example is recommend to use Python 3.9 or newer. Check it:

    python -V
    
  2. Create Virtual Environment

    We suggest to create virtual environments env_itex and install official Tensorflow* in it, like:

    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

Please run your script with Intel® Extension for TensorFlow*.