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
| Container | Framework | Docker Pull Command | Compressed Size |
|---|---|---|---|
| Intel® Extension for PyTorch* Container for Flex Series | PyTorch | docker pull intel-extension-for-pytorch:xpu-flex | 7.04GB |
| Intel® Extension for PyTorch* Container for Max Series | PyTorch | docker pull intel-extension-for-pytorch:xpu-max | 7.13GB |
| Intel® Extension for TensorFlow | TensoFlow | docker pull intel/intel-extension-for-tensorflow:xpu | 2.29GB |
| Intel® Extension for PyTorch* Container for Flex & Max Series with Jupyter notebook support | PyTorch | docker pull intel-extension-for-pytorch:xpu-jupyter | 7.13GB |
| Intel® Extension for TensorFlow with Jupyter notebook support | TensoFlow | docker pull intel/intel-extension-for-tensorflow:xpu-jupyter | 1.88GB |
| Intel® Optimizations for PyTorch for Max Series GPU | PyTorch | docker pull intel/intel-extension-for-pytorch:xpu-max | 2.2GB |
| Intel® Optimizations for PyTorch for Max Series GPU | PyTorch | docker pull intel/intel-extension-for-pytorch:xpu-max-2.0.110-xpu | 2.2GB |
| Intel® Optimizations for PyTorch for Flex Series GPU | PyTorch | docker pull intel/intel-extension-for-pytorch:xpu-flex-2.0.110-xpu | 7.04GB |