Skip to content

Commit 3f4dbd2

Browse files
Add python 3.11 build to GPU CI (#376)
* Add python 3.11 build to GPU CI * Switch to `cupy-cuda11x` Previously CuPy built a wheel for each CUDA `X.Y` version. However the CuPy team later transitioned to CUDA Compatibility, which allows a wheel build to only focus on CUDA major version. So switch to that wheel here. --------- Co-authored-by: jakirkham <[email protected]>
1 parent 8393093 commit 3f4dbd2

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

continuous_integration/gpuci/axis.yaml

+1
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
PYTHON_VER:
22
- "3.9"
33
- "3.10"
4+
- "3.11"
45

56
CUDA_VER:
67
- "11.8.0"

continuous_integration/gpuci/build.sh

+1-1
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@ rapids-logger "Activate conda env"
4141
conda activate dask_image
4242

4343
rapids-logger "Install cupy"
44-
python -m pip install cupy-cuda112 -f https://pip.cupy.dev/pre
44+
python -m pip install cupy-cuda11x -f https://pip.cupy.dev/pre
4545

4646
rapids-logger "Install dask-image"
4747
python -m pip install .

0 commit comments

Comments
 (0)