From 217e5224dbd75fa8c0db0fd8125463b194fed05a Mon Sep 17 00:00:00 2001 From: Sheldon Date: Tue, 17 Oct 2023 11:44:09 +0800 Subject: [PATCH] add gpu support (#27427) add gpu support Signed-off-by: Sheldon --- tests/scripts/values/ci/pr-gpu.yaml | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/tests/scripts/values/ci/pr-gpu.yaml b/tests/scripts/values/ci/pr-gpu.yaml index 1cd1490a9c..2bbff8e0fe 100644 --- a/tests/scripts/values/ci/pr-gpu.yaml +++ b/tests/scripts/values/ci/pr-gpu.yaml @@ -28,10 +28,16 @@ queryCoordinator: queryNode: nodeSelector: nvidia.com/gpu.present: 'true' + extraEnv: + - name: CUDA_VISIBLE_DEVICES + value: "0,1" resources: requests: + nvidia.com/gpu: 2 cpu: "0.5" memory: "500Mi" + limits: + nvidia.com/gpu: 2 indexCoordinator: nodeSelector: @@ -44,10 +50,16 @@ indexCoordinator: indexNode: nodeSelector: nvidia.com/gpu.present: 'true' + extraEnv: + - name: CUDA_VISIBLE_DEVICES + value: "0,1" resources: requests: + nvidia.com/gpu: 2 cpu: "0.5" memory: "500Mi" + limits: + nvidia.com/gpu: 2 dataCoordinator: nodeSelector: