Intel® GPU Base Containers

February 7, 2024 · View on GitHub

Intel® Data Center GPU Max Series and Intel® Data Center GPU Flex Series base containers

ContainerFrameworkDocker Pull CommandCompressed Size
Intel® Extension for PyTorch* Container for Flex SeriesPyTorchdocker pull intel-extension-for-pytorch:xpu-flex7.04GB
Intel® Extension for PyTorch* Container for Max SeriesPyTorchdocker pull intel-extension-for-pytorch:xpu-max7.13GB
Intel® Extension for TensorFlowTensoFlowdocker pull intel/intel-extension-for-tensorflow:xpu2.29GB
Intel® Extension for PyTorch* Container for Flex & Max Series with Jupyter notebook supportPyTorchdocker pull intel-extension-for-pytorch:xpu-jupyter7.13GB
Intel® Extension for TensorFlow with Jupyter notebook supportTensoFlowdocker pull intel/intel-extension-for-tensorflow:xpu-jupyter1.88GB
Intel® Optimizations for PyTorch for Max Series GPUPyTorchdocker pull intel/intel-extension-for-pytorch:xpu-max2.2GB
Intel® Optimizations for PyTorch for Max Series GPUPyTorchdocker pull intel/intel-extension-for-pytorch:xpu-max-2.0.110-xpu2.2GB
Intel® Optimizations for PyTorch for Flex Series GPUPyTorchdocker pull intel/intel-extension-for-pytorch:xpu-flex-2.0.110-xpu7.04GB

Intel AI Containers