Intel® Max (PVC) Model Containers

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

Container	Framework	Docker Pull Command	Compressed Size
ResNet50 v1.5 Inference	PyTorch	docker pull intel/image-recognition:pytorch-max-gpu-resnet50v1-5-inference	2.37GB
ResNet50 v1.5 Training	PyTorch	docker pull intel/image-recognition:pytorch-max-gpu-resnet50v1-5-training	2.51GB
BERT Large Inference	PyTorch	docker pull intel/language-modeling:pytorch-max-gpu-bert-large-inference	7.8GB
BERT Large Training	PyTorch	docker pull intel/language-modeling:pytorch-max-gpu-bert-large-training	2.79GB
Stable Diffusion Inference	PyTorch	docker pull intel/generative-ai:pytorch-max-gpu-stable-diffusion-inference	2.65GB
DistilBERT Inference	PyTorch	docker pull intel/recommendation:pytorch-max-gpu-distilbert-inference	2.54GB
ResNet50 v1.5 Inference	TensorFlow	docker pull intel/image-recognition:tf-max-gpu-resnet50v1-5-inference	2.38GB
ResNet50 v1.5 Training	TensorFlow	docker pull intel/image-recognition:tf-max-gpu-resnet50v1-5-training	3.68GB
BERT Large Inference	TensorFlow	docker pull intel/language-modeling:tf-max-gpu-bert-large-inference	3.33GB
BERT Large Training	TensorFlow	docker pull intel/language-modeling:tf-max-gpu-bert-large-training	2.79GB
DLRM Training	PyTorch	docker pull intel/recommendation:pytorch-max-gpu-dlrm-training	2.98GB
DLRM Inference	PyTorch	docker pull intel/recommendation:pytorch-max-gpu-dlrm-inference	2.91GB
3D-UNet Training	TensorFlow	docker pull intel/image-segmentation:tf-max-gpu-unet-3d-training	2.78GB
Mask R-CNN Training	TensorFlow	docker pull intel/image-segmentation:tf-max-gpu-maskrcnn-training	2.76GB

Intel AI Containers