add gpu support (#27427)

add gpu support

Signed-off-by: Sheldon <chuanfeng.liu@zilliz.com>
This commit is contained in:
Sheldon 2023-10-17 11:44:09 +08:00 committed by GitHub
parent 5dbcdf777d
commit 217e5224db
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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: