Skip to content

Commit

Permalink
Bump GPU CI to CUDA 11.8 (#348)
Browse files Browse the repository at this point in the history
* Bump GPU CI to CUDA 11.8

* Resolve deprecation warnings
  • Loading branch information
charlesbluca authored Nov 28, 2023
1 parent 7c8838b commit 56ec50d
Show file tree
Hide file tree
Showing 2 changed files with 11 additions and 11 deletions.
2 changes: 1 addition & 1 deletion continuous_integration/gpuci/axis.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@ PYTHON_VER:
- "3.10"

CUDA_VER:
- "11.5.2"
- "11.8.0"

LINUX_VER:
- ubuntu20.04
Expand Down
20 changes: 10 additions & 10 deletions continuous_integration/gpuci/build.sh
Original file line number Diff line number Diff line change
Expand Up @@ -27,32 +27,32 @@ export CUDA_REL=${CUDA_VERSION%.*}
# SETUP - Check environment
################################################################################

gpuci_logger "Check environment variables"
rapids-logger "Check environment variables"
env

gpuci_logger "Check GPU usage"
rapids-logger "Check GPU usage"
nvidia-smi

gpuci_logger "Update conda environment"
rapids-logger "Update conda environment"
. /opt/conda/etc/profile.d/conda.sh
gpuci_mamba_retry env update -n dask_image -f "$WORKSPACE/continuous_integration/environment-$PYTHON_VER.yml"
rapids-mamba-retry env update -n dask_image -f "$WORKSPACE/continuous_integration/environment-$PYTHON_VER.yml"

gpuci_logger "Activate conda env"
rapids-logger "Activate conda env"
conda activate dask_image

gpuci_logger "Install cupy"
rapids-logger "Install cupy"
python -m pip install cupy-cuda112 -f https://pip.cupy.dev/pre

gpuci_logger "Install dask-image"
rapids-logger "Install dask-image"
python setup.py install

gpuci_logger "Check Python versions"
rapids-logger "Check Python versions"
python --version

gpuci_logger "Check conda environment"
rapids-logger "Check conda environment"
conda info
conda config --show-sources
conda list --show-channel-urls

gpuci_logger "Python py.test for dask-image"
rapids-logger "Python py.test for dask-image"
py.test $WORKSPACE -v -m cupy --junitxml="$WORKSPACE/junit-dask-image.xml"

0 comments on commit 56ec50d

Please sign in to comment.