Update Docker files to CUDA 12.5 and UCX 1.17

rapidsai · Oct 3, 2024 · 4c3780c · 4c3780c
1 parent 46e8f57
commit 4c3780c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 7 deletions.
diff --git a/docker/UCXPy-rdma-core.dockerfile b/docker/UCXPy-rdma-core.dockerfile
@@ -1,15 +1,15 @@
-ARG CUDA_VERSION=11.5.2
-ARG DISTRIBUTION_VERSION=ubuntu20.04
+ARG CUDA_VERSION=12.5.1
+ARG DISTRIBUTION_VERSION=ubuntu22.04
 FROM nvidia/cuda:${CUDA_VERSION}-devel-${DISTRIBUTION_VERSION}
 
 # Tag to checkout from UCX repository
-ARG UCX_VERSION_TAG=v1.13.0
+ARG UCX_VERSION_TAG=v1.17.0
 # Where to install conda, and what to name the created environment
 ARG CONDA_HOME=/opt/conda
 ARG CONDA_ENV=ucx
 # Name of conda spec file in the current working directory that
 # will be used to build the conda environment.
-ARG CONDA_ENV_SPEC=ucx-py-cuda11.5.yml
+ARG CONDA_ENV_SPEC=ucx-py-cuda12.5.yml
 
 ENV CONDA_ENV="${CONDA_ENV}"
 ENV CONDA_HOME="${CONDA_HOME}"
@@ -32,8 +32,6 @@ RUN apt-get update -y \
         pkg-config \
         udev \
         curl \
-        librdmacm-dev \
-        rdma-core \
     && apt-get autoremove -y \
     && apt-get clean
 

diff --git a/docker/ucx-py-cuda11.5.yml → docker/ucx-py-cuda12.5.yml b/docker/ucx-py-cuda11.5.yml → docker/ucx-py-cuda12.5.yml
@@ -5,7 +5,7 @@ channels:
 
 dependencies:
   - python=3.10
-  - cudatoolkit=11.5
+  - cuda-version=12.5
   - setuptools
   - cython>=3.0.0
   - pytest