Objective

Build a RHEL 9.6 Docker environment for PyTorch compilation and testing.

Work Completed

• Created .ci/docker/rhel9/Dockerfile based on registry.redhat.io/ubi9/ubi:9.6
• Installed Python 3.12 build environment
• Configured GCC 11 compiler
• Added CUDA 12.8 and cuDNN 12.8 support
• Set up conda environment with PyTorch dependencies
• Added compiler flags for FP16 support (-mf16c -mavx2)
• Configured for H200 GPU (compute capability 9.0)

Deliverables

• [x] Docker image builds successfully
• [x] Image size ~20GB
• [x] All dependencies installed
• [x] Build time ~35 minutes

References

• Dockerfile: .ci/docker/rhel9/Dockerfile
• Workflow: .github/workflows/rhel-build-test.yml