Upgrade rapids version for container

NVIDIA-Genomics-Research · Aug 20, 2021 · c4a24ce · c4a24ce
1 parent 89fd524
commit c4a24ce
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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
@@ -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.