From 72721d319eb8ae8fcb2aa76b2d86d791b9beb00e Mon Sep 17 00:00:00 2001
From: Taurean Dyer <46935140+taureandyernv@users.noreply.github.com>
Date: Fri, 15 Nov 2019 09:51:07 -0800
Subject: [PATCH] bandaid for rapids-colab.sh script

changed script to 0.11 compliant, but installs the highest working version, v0.10 stable
---
 utils/rapids-colab.sh | 32 +++++++++++++++++++++++---------
 1 file changed, 23 insertions(+), 9 deletions(-)

diff --git a/utils/rapids-colab.sh b/utils/rapids-colab.sh
index bd7520b4..34ceb0b3 100644
--- a/utils/rapids-colab.sh
+++ b/utils/rapids-colab.sh
@@ -2,8 +2,7 @@
 
 set -eu
 
-RAPIDS_VERSION="${1:-0.11}"
-XGBOOST_VERSION="${2:-1.0.0_SNAPSHOT}"
+RAPIDS_VERSION="${1:-0.10}"
 
 wget -nc https://github.com/rapidsai/notebooks-contrib/raw/master/utils/env-check.py
 echo "Checking for GPU type:"
@@ -23,13 +22,28 @@ if [ ! -f Miniconda3-4.5.4-Linux-x86_64.sh ]; then
     echo "Installing RAPIDS $RAPIDS_VERSION packages"
     echo "Please standby, this will take a few minutes..."
     # install RAPIDS packages
-    conda install -y --prefix /usr/local \
-      -c rapidsai-nightly/label/xgboost -c rapidsai-nightly -c nvidia -c conda-forge \
-      python=3.6 cudatoolkit=10.1 \
-      cudf=$RAPIDS_VERSION cuml cugraph gcsfs pynvml \
-      dask-cudf
-
-    pip install https://xgboost-ci.net/job/xgboost/job/master/lastSuccessfulBuild/artifact/python-package/dist/xgboost-$XGBOOST_VERSION-py2.py3-none-any.whl
+    
+    if [ $RAPIDS_VERSION == "0.11" ] ;then
+    echo "Installing RAPIDS $RAPIDS_VERSION packages from the nightly release channel"
+    echo "Please standby, this will take a few minutes..."
+    # install RAPIDS packages
+        conda install -y --prefix /usr/local \
+                -c rapidsai-nightly/label/xgboost -c rapidsai-nightly -c nvidia -c conda-forge \
+                python=3.6 cudatoolkit=10.1 \
+                cudf=$RAPIDS_VERSION cuml cugraph gcsfs pynvml \
+                dask-cudf \
+                xgboost
+    else
+        echo "Installing RAPIDS $RAPIDS_VERSION packages from the stable release channel"
+        echo "Please standby, this will take a few minutes..."
+        # install RAPIDS packages
+        conda install -y --prefix /usr/local \
+            -c rapidsai/label/xgboost -c rapidsai -c nvidia -c conda-forge \
+            python=3.6 cudatoolkit=10.1 \
+            cudf=$RAPIDS_VERSION cuml cugraph gcsfs pynvml \
+            dask-cudf \
+            xgboost
+    fi
       
     echo "Copying shared object files to /usr/lib"
     # copy .so files to /usr/lib, where Colab's Python looks for libs