From 9e90ccd5e82e2ccd98f588e47a327579678b996d Mon Sep 17 00:00:00 2001 From: James Lamb Date: Tue, 30 Jul 2024 10:25:48 -0500 Subject: [PATCH 1/2] Update pip devcontainers to UCX v1.17.0 --- .devcontainer/cuda11.8-pip/devcontainer.json | 2 +- .devcontainer/cuda12.5-pip/devcontainer.json | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/.devcontainer/cuda11.8-pip/devcontainer.json b/.devcontainer/cuda11.8-pip/devcontainer.json index 146753ae0..ed4656785 100644 --- a/.devcontainer/cuda11.8-pip/devcontainer.json +++ b/.devcontainer/cuda11.8-pip/devcontainer.json @@ -16,7 +16,7 @@ "hostRequirements": {"gpu": "optional"}, "features": { "ghcr.io/rapidsai/devcontainers/features/ucx:24.10": { - "version": "1.15.0" + "version": "1.17.0" }, "ghcr.io/rapidsai/devcontainers/features/cuda:24.10": { "version": "11.8", diff --git a/.devcontainer/cuda12.5-pip/devcontainer.json b/.devcontainer/cuda12.5-pip/devcontainer.json index 49e4943eb..b05b14bf6 100644 --- a/.devcontainer/cuda12.5-pip/devcontainer.json +++ b/.devcontainer/cuda12.5-pip/devcontainer.json @@ -16,7 +16,7 @@ "hostRequirements": {"gpu": "optional"}, "features": { "ghcr.io/rapidsai/devcontainers/features/ucx:24.10": { - "version": "1.15.0" + "version": "1.17.0" }, "ghcr.io/rapidsai/devcontainers/features/cuda:24.10": { "version": "12.5", From 04313a3c612f48968c13ee90bbfee07585423dd7 Mon Sep 17 00:00:00 2001 From: James Lamb Date: Tue, 30 Jul 2024 10:41:26 -0500 Subject: [PATCH 2/2] just use images with prebuilt ucx --- .devcontainer/cuda11.8-pip/devcontainer.json | 5 +---- .devcontainer/cuda12.5-pip/devcontainer.json | 5 +---- 2 files changed, 2 insertions(+), 8 deletions(-) diff --git a/.devcontainer/cuda11.8-pip/devcontainer.json b/.devcontainer/cuda11.8-pip/devcontainer.json index ed4656785..74d62afcc 100644 --- a/.devcontainer/cuda11.8-pip/devcontainer.json +++ b/.devcontainer/cuda11.8-pip/devcontainer.json @@ -5,7 +5,7 @@ "args": { "CUDA": "11.8", "PYTHON_PACKAGE_MANAGER": "pip", - "BASE": "rapidsai/devcontainers:24.10-cpp-cuda11.8-ubuntu22.04" + "BASE": "rapidsai/devcontainers:24.10-cpp-cuda11.8-ucx1.17.0-openmpi-ubuntu22.04" } }, "runArgs": [ @@ -15,9 +15,6 @@ ], "hostRequirements": {"gpu": "optional"}, "features": { - "ghcr.io/rapidsai/devcontainers/features/ucx:24.10": { - "version": "1.17.0" - }, "ghcr.io/rapidsai/devcontainers/features/cuda:24.10": { "version": "11.8", "installcuBLAS": true, diff --git a/.devcontainer/cuda12.5-pip/devcontainer.json b/.devcontainer/cuda12.5-pip/devcontainer.json index b05b14bf6..3dcf52e83 100644 --- a/.devcontainer/cuda12.5-pip/devcontainer.json +++ b/.devcontainer/cuda12.5-pip/devcontainer.json @@ -5,7 +5,7 @@ "args": { "CUDA": "12.5", "PYTHON_PACKAGE_MANAGER": "pip", - "BASE": "rapidsai/devcontainers:24.10-cpp-cuda12.5-ubuntu22.04" + "BASE": "rapidsai/devcontainers:24.10-cpp-cuda12.5-ucx1.17.0-openmpi-ubuntu22.04" } }, "runArgs": [ @@ -15,9 +15,6 @@ ], "hostRequirements": {"gpu": "optional"}, "features": { - "ghcr.io/rapidsai/devcontainers/features/ucx:24.10": { - "version": "1.17.0" - }, "ghcr.io/rapidsai/devcontainers/features/cuda:24.10": { "version": "12.5", "installcuBLAS": true,