Intel® Flex (ATSM) Model Containers

These are the deep learning Models optimized to run on Intel's Flex Series GPU platform.

Container	Framework	Docker Pull Command	Compressed Size
ResNet50 v1.5 Inference	TensorFlow	docker pull intel/image-recognition:tf-flex-gpu-resnet50v1-5-inference	2.35GB
MaskRCNN Inference	TensorFlow	docker pull intel/image-segmentation:tf-flex-gpu-maskrcnn-inference	2.59GB
EfficientNet Inference	TensorFlow	docker pull intel/image-recognition:tf-flex-gpu-efficientnet-inference	2.3GB
Stable Diffusion Inference	TensorFlow	docker pull intel/generative-ai:tf-flex-gpu-stable-diffusion-inference	2.54GB
Wide and Deep Inference	TensorFlow	docker pull intel/recommendation:tf-flex-gpu-wide-and-deep-inference	2.33GB
ResNet50 v1.5 Inference	PyTorch	docker pull intel/image-recognition:pytorch-flex-gpu-resnet50v1-5-inference	2.17GB
YOLOv5 inference	PyTorch	docker pull intel/object-detection:pytorch-flex-gpu-yolov5-inference	2.54GB
Stable Diffusion inference	PyTorch	docker pull intel/generative-ai:pytorch-flex-gpu-stable-diffusion-inference	2.65GB
DistilBERT inference	PyTorch	docker pull intel/language-modeling:pytorch-flex-gpu-distilbert-inference	2.54GB
DLRM-v1 inference	PyTorch	docker pull intel/recommendation:pytorch-flex-gpu-dlrm-v1-inference	2.58GB
SSD-MobileNet v1 Inference	PyTorch	docker pull intel/object-detection:pytorch-flex-gpu-ssd-mobilenet-inference	2.47GB
YOLOv4 inference	PyTorch	docker pull intel/object-detection:pytorch-flex-gpu-yolov4-inference	2.56GB
SSD-MobileNet v1 Inference	TensorFlow	docker pull intel/object-detection:tf-flex-gpu-ssd-mobilenet-inference	2.47GB

Intel AI Containers