diff --git a/test/images/cuda-vector-add/BASEIMAGE b/test/images/cuda-vector-add/BASEIMAGE index 5e5cd4dac28..a89f93aef66 100644 --- a/test/images/cuda-vector-add/BASEIMAGE +++ b/test/images/cuda-vector-add/BASEIMAGE @@ -1,2 +1,3 @@ -linux/amd64=nvidia/cuda:10.0-devel-ubuntu18.04 -linux/ppc64le=nvidia/cuda-ppc64le:10.0-devel-ubuntu18.04 +linux/amd64=nvidia/cuda:11.1.1-devel-ubuntu18.04 +linux/arm64=nvidia/cuda:11.1.1-devel-ubuntu18.04 +linux/ppc64le=nvidia/cuda:11.1.1-devel-ubuntu18.04 diff --git a/test/images/cuda-vector-add/Dockerfile b/test/images/cuda-vector-add/Dockerfile index af82e6a5e9b..c2c6d10a5f2 100644 --- a/test/images/cuda-vector-add/Dockerfile +++ b/test/images/cuda-vector-add/Dockerfile @@ -15,8 +15,8 @@ ARG BASEIMAGE FROM $BASEIMAGE +ENV CUDA_PKG_VERSION=11-1 CROSS_BUILD_COPY qemu-QEMUARCH-static /usr/bin/ - RUN apt-get update && apt-get install -y --no-install-recommends \ cuda-samples-$CUDA_PKG_VERSION && \ rm -rf /var/lib/apt/lists/* diff --git a/test/images/cuda-vector-add/VERSION b/test/images/cuda-vector-add/VERSION index 8bbe6cf74a1..bb576dbde10 100644 --- a/test/images/cuda-vector-add/VERSION +++ b/test/images/cuda-vector-add/VERSION @@ -1 +1 @@ -2.2 +2.3