From 7c808e762eb73de09900fe2d692d6b0c5e80d6f2 Mon Sep 17 00:00:00 2001
From: Ray Douglass <3107146+raydouglass@users.noreply.github.com>
Date: Thu, 1 Feb 2024 09:55:56 -0500
Subject: [PATCH] Pin cuda-python (#624)

Fixes attempted in https://github.com/rapidsai/integration/pull/695 & https://github.com/rapidsai/integration/pull/697 don't play well with CEC at large, but this repo can pin `cuda-python` more restrictively.

This PR pins `cuda-python` to the CUDA `major.minor.*` version.

See also https://github.com/conda-forge/cuda-python-feedstock/issues/66

Authors:
  - Ray Douglass (https://github.com/raydouglass)

Approvers:
  - Bradley Dice (https://github.com/bdice)
  - Jake Awe (https://github.com/AyodeAwe)

URL: https://github.com/rapidsai/docker/pull/624
---
 Dockerfile | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/Dockerfile b/Dockerfile
index 0017cb27..802f40f0 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -45,11 +45,13 @@ WORKDIR /home/rapids
 
 COPY condarc /opt/conda/.condarc
 
+# cuda-python pin required for CUDA 12. See https://github.com/conda-forge/cuda-python-feedstock/issues/66
 RUN <<EOF
 mamba install -y -n base \
     "rapids=${RAPIDS_VER}.*" \
     "dask-sql=${DASK_SQL_VER%.*}.*" \
     "python=${PYTHON_VER}.*" \
+    "cuda-python=${CUDA_VER%.*}.*" \
     "cuda-version=${CUDA_VER%.*}.*" \
     ipython
 conda clean -afy