Intel® CPU Model Containers

These are the deep learning Models optimized to run on Intel® Xeon® Scalable processors

Container	Framework	Docker Pull Command	Compressed Size
ResNet 50v1.5 Training	TensorFlow	docker pull intel/image-recognition:centos-tf-cpu-resnet50v1-5-training	1.46GB
ResNet 50v1.5 Inference	TensorFlow	docker pull intel/image-recognition:centos-tf-cpu-resnet50v1-5-inference	1.69GB
BERT large Training	TensorFlow	docker pull intel/language-modeling:centos-tf-cpu-bert-large-pretraining	1.46GB
BERT large Inference	TensorFlow	docker pull intel/language-modeling:centos-tf-cpu-bert-large-inference	6.76GB
DIEN Inference	TensorFlow	docker pull intel/recommendation:centos-tf-cpu-dien-inference	1.61GB
MobileNet V1* Inference	TensorFlow	docker pull intel/image-recognition:centos-tf-cpu-mobilenet-v1-inference	1.39GB
3D U-Net MLPerf* Inference	TensorFlow	docker pull intel/image-segmentation:centos-tf-cpu-3d-unet-mlperf-inference	2.41GB
SSD-ResNet34 Training	TensorFlow	docker pull intel/object-detection:centos-tf-cpu-ssd-resnet34-training	2.83GB
SSD-ResNet34 Inference	TensorFlow	docker pull intel/object-detection:centos-tf-cpu-ssd-resnet34-inference	2.9GB
SSD-MobileNet* Inference	TensorFlow	docker pull intel/object-detection:centos-tf-cpu-ssd-mobilenet-inference	2.56GB
Transformer_LT_mlperf Training	TensorFlow	docker pull intel/language-translation:centos-tf-cpu-transformer-mlperf-training	1.46GB
Transformer_LT_mlperf Inference	TensorFlow	docker pull intel/language-translation:centos-tf-cpu-transformer-mlperf-inference	3.1GB
ResNet50 Training	PyTorch	docker pull intel/image-recognition:centos-pytorch-cpu-resnet50-training	1.63GB
ResNet50 Inference	PyTorch	docker pull intel/image-recognition:centos-pytorch-cpu-resnet50-inference	1.58GB
ResNext-32x16d Inference	PyTorch	docker pull intel/image-recognition:centos-pytorch-cpu-resnext-32x16d-inference	1.58GB
Mask R-CNN Training	PyTorch	docker pull intel/object-detection:centos-pytorch-cpu-maskrcnn-training	1.76GB
Mask R-CNN Inference	PyTorch	docker pull intel/object-detection:centos-pytorch-cpu-maskrcnn-inference	1.47GB
SSD-ResNet34 Training	PyTorch	docker pull intel/object-detection:centos-pytorch-cpu-ssd-resnet34-training	1.79GB
SSD-ResNet34 Inference	PyTorch	docker pull intel/object-detection:centos-pytorch-cpu-ssd-resnet34-inference	1.76GB
BERT large Training	PyTorch	docker pull intel/language-modeling:centos-pytorch-cpu-bert-large-training	1.99GB
BERT large Inference	PyTorch	docker pull intel/language-modeling:centos-pytorch-cpu-bert-large-inference	1.82GB
DistilBERT base Inference	PyTorch	docker pull intel/language-modeling:centos-pytorch-cpu-distilbert-inference	2.04GB
RNNT Training	PyTorch	docker pull intel/language-modeling:centos-pytorch-cpu-rnnt-training	1.9GB
RNNT Inference	PyTorch	docker pull intel/language-modeling:centos-pytorch-cpu-rnnt-inference	1.85GB
DLRM Training	PyTorch	docker pull intel/recommendation:centos-pytorch-cpu-dlrm-training	1.57GB
DLRM Inference	PyTorch	docker pull intel/recommendation:centos-pytorch-cpu-dlrm-inference	1.53GB

Intel AI Containers