PyTorch* on 3rd Gen Intel® Xeon® Scalable Processors


Pull Command

docker pull intel/image-recognition:pytorch-1.5.0-rc3-icx-a37fb5e8

Tags & Pull Commands for Other Versions

OS Target Version Size Updated Pull Command
Linux x86_64 1.5 2.89GB 2/2021 docker pull intel/recommendation:pytorch-1.5.0-rc3-icx-a37fb5e8
Linux x86_64 1.5 2.89GB 2/2021 docker pull intel/recommendation:pytorch-1.5.0-rc3-icx-a37fb5e8-dlrm-int8
Linux x86_64 1.5 2.89GB 2/2021 docker pull intel/recommendation:pytorch-1.5.0-rc3-icx-a37fb5e8-dlrm-fp32
Linux x86_64 1.5 2.87GB 2/2021 docker pull intel/image-recognition:pytorch-1.5.0-rc3-icx-a37fb5e8-resnet50-int8
Linux x86_64 1.5 2.87GB 2/2021 docker pull intel/image-recognition:pytorch-1.5.0-rc3-icx-a37fb5e8-resnet50-fp32


These images are for customer preview only. Optimizations in these containers will be available officially in the next release of Intel® Extensions for PyTorch*.

3rd Gen Intel® Xeon® Scalable processors, code-named Ice Lake, deliver industry-leading, workload-optimized platforms with built-in AI acceleration, providing a seamless performance foundation to help speed data’s transformative impact. Enhanced Intel Deep Learning Boost, with the industry’s first x86 support of Brain Floating Point 16-bit (blfoat16) numeric format and Vector Neural Network Instructions (VNNI), brings enhanced artificial intelligence inference and training performance, with up to 1.93X more AI training performance and 1.87X more AI inference performance for image classification vs. the prior generation.

Intel® Extensions for PyTorch* extends the original PyTorch* framework by creating extensions that optimize performance of Deep Learning models. This container contains PyTorch* v1.5.0 and Intel® Extensions for Pytorch* version 0.2.

Documentation and Sources

Get Started
GitHub Repo
Get Started

Code Sources


Legal Notice

By accessing, downloading or using this software and any required dependent software (the “Software Package”), you agree to the terms and conditions of the software license agreements for the Software Package, which may also include notices, disclaimers, or license terms for third party software included with the Software Package. Please refer to the license file for additional details.

Related Containers and Solutions

Intel® Extension for PyTorch*
ResNet50 FP32 Inference PyTorch* Container

View All Containers and Solutions 🡢

Product and Performance Information


Performance varies by use, configuration and other factors. Learn more at