From 7520c959c7ddf08fb6f57b8db98be681f60ca873 Mon Sep 17 00:00:00 2001 From: Sebastian Fischer Date: Thu, 26 Mar 2026 09:34:06 +0100 Subject: [PATCH] Add gpytorch to benchmark Docker images Needed for the new GP benchmark that compares anvil vs gpytorch. Co-Authored-By: Claude Opus 4.6 (1M context) --- cpu-bench/Dockerfile | 3 ++- cuda-bench/Dockerfile | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/cpu-bench/Dockerfile b/cpu-bench/Dockerfile index 47220ce..03062e6 100644 --- a/cpu-bench/Dockerfile +++ b/cpu-bench/Dockerfile @@ -10,7 +10,8 @@ RUN apt-get update \ # --break-system-packages is needed because Ubuntu 24.04+ marks system Python as # externally managed (PEP 668). Safe here since this is a container. RUN pip3 install --no-cache-dir --break-system-packages \ - torch --index-url https://download.pytorch.org/whl/cpu + torch --index-url https://download.pytorch.org/whl/cpu \ + && pip3 install --no-cache-dir --break-system-packages gpytorch # Use pppm COPY .Rprofile /usr/local/lib/R/etc/Rprofile.site diff --git a/cuda-bench/Dockerfile b/cuda-bench/Dockerfile index cb77a41..3d84c50 100644 --- a/cuda-bench/Dockerfile +++ b/cuda-bench/Dockerfile @@ -10,7 +10,8 @@ RUN apt-get update \ # --break-system-packages is needed because Ubuntu 24.04+ marks system Python as # externally managed (PEP 668). Safe here since this is a container. RUN pip3 install --no-cache-dir --break-system-packages \ - torch --index-url https://download.pytorch.org/whl/cu124 + torch --index-url https://download.pytorch.org/whl/cu124 \ + && pip3 install --no-cache-dir --break-system-packages gpytorch # Use pppm COPY .Rprofile /usr/local/lib/R/etc/Rprofile.site