use CMD instead of RUN in GPU tests to make env ready when running tests

aws · Dec 24, 2024 · a8af79a · a8af79a
1 parent a7a8447
commit a8af79a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
+CMD python3 cuda_validation.py
diff --git 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
+CMD python3 cuda_validation.py