Use docker

Notes

If you use linux kernerl under version 5.4 in host, upgrade it.

How to build an image

Run the following commands to build a pip based container with the latest stable version prebuilt wheel files:

$ cd $DOCKERFILE_DIR
$ DOCKER_BUILDKIT=1 docker build -f Dockerfile.prebuilt -t intel-extension-for-pytorch:main .

Run the following commands to build a conda based container with Intel® Extension for PyTorch* compiled from source:

$ git clone https://github.com/intel/intel-extension-for-pytorch.git
$ cd intel-extension-for-pytorch
$ git submodule sync
$ git submodule update --init --recursive
$ DOCKER_BUILDKIT=1 docker build -f docker/Dockerfile.compile -t intel-extension-for-pytorch:main .

Sanity Test

When a docker image is built out, Run the command below to launch into a container:

$ docker run --rm -it intel-extension-for-pytorch:main bash

Then run the command below inside the container to verify correct installation.

# python -c "import torch; import intel_extension_for_pytorch as ipex; print('torch:', torch.__version__,' ipex: ',ipex.__version__)"