Update Dockerfile

[KeitaW]

Every stage has individual args, hence the following arguments

ARG GDRCOPY_VERSION=v2.4.1
ARG EFA_INSTALLER_VERSION=1.33.0
ARG AWS_OFI_NCCL_VERSION=v1.9.2-aws
ARG NCCL_VERSION=v2.21.5-1
ARG NCCL_TESTS_VERSION=v2.13.9

cannot be retrieved under the FROM statement:

#6 [ 2/19] RUN echo "Versions: CUDA=12.2.2 GDDRCOPY=${GDRCOPY_VERSION} EFA=${EFA_INSTALLER_VERSION} OFI=${AWS_OFI_NCCL_VERSION} NCCL=2.19.3-1 TESTS=${NCCL_TESTS_VERSION}"
#6 0.212 Versions: CUDA=12.2.2 GDDRCOPY= EFA= OFI= NCCL=2.19.3-1 TESTS=

this PR is suggest to put them right after the FROM statement.

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

[KeitaW]

docker/cli#333 (comment)

[iankouls-aws]

This looks good. Only CUDA_VERSION needs to be above the FROM line

[KeitaW]

Thank you!