From 411df9f878081ae8d021f86663ac6b28f72348b3 Mon Sep 17 00:00:00 2001 From: Philip Hyunsu Cho Date: Thu, 1 Nov 2018 08:34:47 -0700 Subject: [PATCH] Test wheels on CUDA 10.0 container for compatibility (#3838) --- Jenkinsfile | 6 +++++- tests/ci_build/Dockerfile.gpu | 5 +++-- 2 files changed, 8 insertions(+), 3 deletions(-) diff --git a/Jenkinsfile b/Jenkinsfile index 23d28fc7eb95..6ef117581216 100644 --- a/Jenkinsfile +++ b/Jenkinsfile @@ -96,7 +96,11 @@ def buildPlatformCmake(buildName, conf, nodeReq, dockerTarget) { # Test the wheel for compatibility on a barebones CPU container ${dockerRun} release ${dockerArgs} bash -c " \ pip install --user python-package/dist/xgboost-*-none-any.whl && \ - python -m nose tests/python" + python -m nose -v tests/python" + # Test the wheel for compatibility on CUDA 10.0 container + ${dockerRun} gpu --build-arg CUDA_VERSION=10.0 bash -c " \ + pip install --user python-package/dist/xgboost-*-none-any.whl && \ + python -m nose -v --eval-attr='(not slow) and (not mgpu)' tests/python-gpu" """ } } diff --git a/tests/ci_build/Dockerfile.gpu b/tests/ci_build/Dockerfile.gpu index 3805a66f8caf..c64acba0ed13 100644 --- a/tests/ci_build/Dockerfile.gpu +++ b/tests/ci_build/Dockerfile.gpu @@ -21,13 +21,14 @@ RUN \ # NCCL2 (License: https://docs.nvidia.com/deeplearning/sdk/nccl-sla/index.html) RUN \ - export CUDA_SHORT=`echo $CUDA_VERSION | egrep -o '[0-9]\.[0-9]'` && \ + export CUDA_SHORT=`echo $CUDA_VERSION | egrep -o '[0-9]+\.[0-9]'` && \ + if [ "${CUDA_SHORT}" != "10.0" ]; then \ wget https://developer.download.nvidia.com/compute/redist/nccl/v2.2/nccl_2.2.13-1%2Bcuda${CUDA_SHORT}_x86_64.txz && \ tar xf "nccl_2.2.13-1+cuda${CUDA_SHORT}_x86_64.txz" && \ cp nccl_2.2.13-1+cuda${CUDA_SHORT}_x86_64/include/nccl.h /usr/include && \ cp nccl_2.2.13-1+cuda${CUDA_SHORT}_x86_64/lib/* /usr/lib && \ rm -f nccl_2.2.13-1+cuda${CUDA_SHORT}_x86_64.txz && \ - rm -r nccl_2.2.13-1+cuda${CUDA_SHORT}_x86_64 + rm -r nccl_2.2.13-1+cuda${CUDA_SHORT}_x86_64; fi ENV PATH=/opt/python/bin:$PATH ENV CC=/opt/rh/devtoolset-2/root/usr/bin/gcc