Preview PyTorch* on 3rd-Generation Intel® Xeon® Scalable Processors

Published: 04/08/2021

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.89 GB 2/2021 docker pull intel/recommendation:pytorch-1.5.0-rc3-icx-a37fb5e8
Linux* x86_64 1.5 2.89 GB 2/2021 docker pull intel/recommendation:pytorch-1.5.0-rc3-icx-a37fb5e8-dlrm-int8
Linux* x86_64 1.5 2.89 GB 2/2021 docker pull intel/recommendation:pytorch-1.5.0-rc3-icx-a37fb5e8-dlrm-fp32
Linux* x86_64 1.5 2.87 GB 2/2021 docker pull intel/image-recognition:pytorch-1.5.0-rc3-icx-a37fb5e8-resnet50-int8
Linux* x86_64 1.5 2.87 GB 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® Extension for PyTorch*.

3rd generation 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 AI 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® Extension 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® Extension for Pytorch* version 0.2.

Documentation and Sources

Get Started
Docker Hub*
GitHub* Repository
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