From a8af79a10548ce7361dcc5dbd0d1b3e2c27448c7 Mon Sep 17 00:00:00 2001 From: Ruinong Tian Date: Tue, 24 Dec 2024 17:57:05 +0000 Subject: [PATCH] use CMD instead of RUN in GPU tests to make env ready when running tests --- test/test_artifacts/v0/gpu-dependencies.test.Dockerfile | 2 +- test/test_artifacts/v1/gpu-dependencies.test.Dockerfile | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/test/test_artifacts/v0/gpu-dependencies.test.Dockerfile b/test/test_artifacts/v0/gpu-dependencies.test.Dockerfile index 2e48e09d..f008e6fc 100644 --- a/test/test_artifacts/v0/gpu-dependencies.test.Dockerfile +++ b/test/test_artifacts/v0/gpu-dependencies.test.Dockerfile @@ -8,4 +8,4 @@ ARG MAMBA_DOCKERFILE_ACTIVATE=1 # 2. Check if Pytorch is installed with CUDA support for GPU image COPY --chown=$MAMBA_USER:$MAMBA_USER scripts/cuda_validation.py . RUN chmod +x cuda_validation.py -RUN python3 cuda_validation.py \ No newline at end of file +CMD python3 cuda_validation.py \ No newline at end of file diff --git a/test/test_artifacts/v1/gpu-dependencies.test.Dockerfile b/test/test_artifacts/v1/gpu-dependencies.test.Dockerfile index 2e48e09d..f008e6fc 100644 --- a/test/test_artifacts/v1/gpu-dependencies.test.Dockerfile +++ b/test/test_artifacts/v1/gpu-dependencies.test.Dockerfile @@ -8,4 +8,4 @@ ARG MAMBA_DOCKERFILE_ACTIVATE=1 # 2. Check if Pytorch is installed with CUDA support for GPU image COPY --chown=$MAMBA_USER:$MAMBA_USER scripts/cuda_validation.py . RUN chmod +x cuda_validation.py -RUN python3 cuda_validation.py \ No newline at end of file +CMD python3 cuda_validation.py \ No newline at end of file