Skip to content

Latest commit

 

History

History

docker

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
Dockerfile.compile
Dockerfile.compile
 
 
Dockerfile.prebuilt
Dockerfile.prebuilt
 
 
README.md
README.md
 
 

README.md

Use docker

  • Notes

    If you use linux kernel 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__)"