use CMD instead of RUN in GPU test, to make it run in proper env

aws · Nov 15, 2024 · 07e5e2c · 07e5e2c
1 parent 01117d5
commit 07e5e2c
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