From c4a24ce8c4c94a1fed9cba5ffd7b619e4049e67e Mon Sep 17 00:00:00 2001
From: Rajesh Ilango <rilango@nvidia.com>
Date: Fri, 20 Aug 2021 12:16:48 -0700
Subject: [PATCH] Upgrade rapids version for container

---
 Dockerfile | 2 +-
 build.sh   | 2 +-
 launch     | 2 +-
 3 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index a9cf628..796c612 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
-ARG BASE_IMAGE=rapidsai/rapidsai:cuda11.0-runtime-ubuntu18.04-py3.7
+ARG BASE_IMAGE=rapidsai/rapidsai:21.06-cuda11.0-runtime-ubuntu18.04-py3.7
 
 FROM ${BASE_IMAGE}
 RUN apt update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
diff --git a/build.sh b/build.sh
index d7ed90d..77d2155 100755
--- a/build.sh
+++ b/build.sh
@@ -29,7 +29,7 @@ set -e
 
 BASE_DIR=$(dirname $0)
 IMAGE_NAME=claraparabricks/single-cell-examples_rapids_cuda11.0
-BASE_IMAGE=rapidsai/rapidsai:0.17-cuda11.0-runtime-ubuntu18.04-py3.7
+BASE_IMAGE=rapidsai/rapidsai:21.06-cuda11.0-runtime-ubuntu18.04-py3.7
 GIT_BRANCH='master'
 CONTAINER_TAG='v0.0.4'
 HELP=false
diff --git a/launch b/launch
index 4c555dd..0613427 100755
--- a/launch
+++ b/launch
@@ -12,7 +12,7 @@ from distutils.version import StrictVersion
 from subprocess import run, Popen
 
 
-DOCKER_IMAGE = 'claraparabricks/single-cell-examples_rapids_cuda11.0:v0.0.3'
+DOCKER_IMAGE = 'claraparabricks/single-cell-examples_rapids_cuda11.0:v0.0.4'
 
 # Key: String: Example name
 # Value: String: Comma seperated HTTP URL to dataset files.