You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
HIP_VISIBLE_DEVICES - this variable specifies the ID of the GPU that you want to use.
You can specify either one or several comma-separated ones - "0" or "0,1,2,3"
TENSOR_PARALLEL_SIZE - must match the number of GPUs used
resources:
limits:
amd.com/gpu: "1" - replace "1" with the number of GPUs used
HIP_VISIBLE_DEVICES - this variable specifies the ID of the GPU that you want to use.
You can specify either one or several comma-separated ones - "0" or "0,1,2,3"
extraCmdArgs: [ "--num-shard","1" ] - replace "1" with the number of GPUs used
resources:
limits:
amd.com/gpu: "1" - replace "1" with the number of GPUs used
HIP_VISIBLE_DEVICES - this variable specifies the ID of the GPU that you want to use.
You can specify either one or several comma-separated ones - "0" or "0,1,2,3"
TENSOR_PARALLEL_SIZE - must match the number of GPUs used
resources:
limits:
amd.com/gpu: "1" - replace "1" with the number of GPUs used
HIP_VISIBLE_DEVICES - this variable specifies the ID of the GPU that you want to use.
You can specify either one or several comma-separated ones - "0" or "0,1,2,3"
extraCmdArgs: [ "--num-shard","1" ] - replace "1" with the number of GPUs used
resources:
limits:
amd.com/gpu: "1" - replace "1" with the number of GPUs used