Test wheels on CUDA 10.0 container for compatibility (#3838)

2018-11-01 08:34:47 -07:00
parent 42200ec03e
commit 411df9f878
2 changed files with 8 additions and 3 deletions
--- a/6
+++ b/6
@@ -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"
                """
            }
        }