diff --git a/Dockerfile b/Dockerfile index 2eaa47cdc..474be338c 100644 --- a/Dockerfile +++ b/Dockerfile @@ -3,9 +3,9 @@ ARG VARIANT # Pre-stages for base image variants -FROM ${BASE_IMAGE} as base-default +FROM ${BASE_IMAGE} AS base-default -FROM ${BASE_IMAGE} as base-cuda +FROM ${BASE_IMAGE} AS base-cuda ARG CUDA_VERSION