Pin libcusparse and libcusolver to avoid CUDA 12 (#1095)

Authors:
  - William Hicks (https://github.com/wphicks)
  - Dante Gama Dessavre (https://github.com/dantegd)

Approvers:
  - Corey J. Nolet (https://github.com/cjnolet)
  - Ray Douglass (https://github.com/raydouglass)

URL: #1095

ci/gpu/build.sh

@@ -34,7 +34,7 @@ unset GIT_DESCRIBE_TAG
 export UCX_PY_VERSION='0.30.*'
 
 # Whether to install dask nightly or stable packages.
-export INSTALL_DASK_MAIN=0
+export INSTALL_DASK_MAIN=1
 
 # Dask version to install when `INSTALL_DASK_MAIN=0`
 export DASK_STABLE_VERSION="2022.11.1"

conda/environments/all_cuda-115_arch-x86_64.yaml

@@ -17,8 +17,8 @@ dependencies:
 - cxx-compiler
 - cython>=0.29,<0.30
 - dask-cuda=23.02.*
-- dask==2022.11.1
-- distributed==2022.11.1
+- dask>=2022.12.0
+- distributed>=2022.12.0
 - doxygen>=1.8.20
 - faiss-proc=*=cuda
 - gcc_linux-64=9.*

conda/recipes/libraft/conda_build_config.yaml

@@ -20,10 +20,10 @@ gtest_version:
   - "=1.10.0"
 
 libcusolver_version:
-  - ">=11.2.1"
+  - ">=11.2.1,<=11.4.1.48"
 
 libcusparse_version:
-  - ">=11.5.0"
+  - ">=11.5.0,<12.0"
 
 libfaiss_version:
   - "1.7.0 *_cuda"

conda/recipes/raft-dask/meta.yaml

@@ -51,8 +51,8 @@ requirements:
     - ucx >={{ ucx_version }}
     - ucx-py {{ ucx_py_version }}
     - ucx-proc=*=gpu
-    - dask==2022.11.1
-    - distributed==2022.11.1
+    - dask>=2022.12.0
+    - distributed>=2022.12.0
     - cuda-python >=11.7.1,<12.0
     - joblib >=0.11
     - {{ pin_compatible('cudatoolkit', max_pin='x', min_pin='x') }}

dependencies.yaml

@@ -109,8 +109,8 @@ dependencies:
       - output_types: [conda]
         packages:
           - rmm=23.02.*
-          - dask==2022.11.1
-          - distributed==2022.11.1
+          - dask>=2022.12.0
+          - distributed>=2022.12.0
           - ucx>=1.13.0
           - ucx-py=0.30.*
           - ucx-proc=*=gpu