diff --git a/src/pages/resources/components/add-worker.tsx b/src/pages/resources/components/add-worker.tsx index b5e3322f..fd0cbc76 100644 --- a/src/pages/resources/components/add-worker.tsx +++ b/src/pages/resources/components/add-worker.tsx @@ -38,7 +38,7 @@ const AddWorker: React.FC = (props) => { closeIcon={true} maskClosable={false} keyboard={false} - width={600} + width={700} styles={{ body: { height: 550 diff --git a/src/pages/resources/components/container-install.tsx b/src/pages/resources/components/container-install.tsx index 2de53ff0..2a737537 100644 --- a/src/pages/resources/components/container-install.tsx +++ b/src/pages/resources/components/container-install.tsx @@ -83,6 +83,7 @@ const AddWorker: React.FC = (props) => {

2. {intl.formatMessage({ id: 'resources.worker.add.step2' })}{' '} @@ -116,7 +117,7 @@ const AddWorker: React.FC = (props) => { }} > )} - +

3. {intl.formatMessage({ id: 'resources.worker.add.step3' })}

diff --git a/src/pages/resources/components/styles/installation.less b/src/pages/resources/components/styles/installation.less index 8d0e2ee9..2bd04246 100644 --- a/src/pages/resources/components/styles/installation.less +++ b/src/pages/resources/components/styles/installation.less @@ -19,6 +19,10 @@ .code-pre { white-space: pre-wrap; } + + .ant-radio-button-wrapper { + flex: auto; + } } .script-install { diff --git a/src/pages/resources/config/index.ts b/src/pages/resources/config/index.ts index 68f2cdef..d8dbef26 100644 --- a/src/pages/resources/config/index.ts +++ b/src/pages/resources/config/index.ts @@ -90,6 +90,16 @@ export const addWorkerGuide: Record = { return `docker run -d --name gpustack-worker --restart=unless-stopped -p 10150:10150 -p 40000-41024:40000-41024 -p 50000-51024:50000-51024 --ipc=host --group-add=video --security-opt seccomp=unconfined --device /dev/kfd --device /dev/dri -v gpustack-worker-data:/var/lib/gpustack gpustack/gpustack:${params.tag} --server-url ${params.server} --token ${params.token} --worker-ip ${params.workerip}`; } }, + dcu: { + registerWorker(params: { + server: string; + tag: string; + token: string; + workerip: string; + }) { + return `docker run -itd --shm-size 500g --network=host --privileged --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined --device=/dev/kfd --device=/dev/dri -v /opt/hyhal:/opt/hyhal:ro gpustack/gpustack:${params.tag} --server-url ${params.server} --token ${params.token} --worker-ip ${params.workerip}`; + } + }, container: { getToken: 'docker exec -it ${gpustack_container_id} cat /var/lib/gpustack/token' @@ -97,9 +107,10 @@ export const addWorkerGuide: Record = { }; export const containerInstallOptions = [ - { label: 'CUDA', value: 'cuda' }, - { label: 'AMD', value: 'rocm' }, - { label: 'CANN', value: 'npu' }, - { label: 'MUSA', value: 'musa' }, + { label: 'NVIDIA CUDA', value: 'cuda' }, + { label: 'AMD ROCm', value: 'rocm' }, + { label: 'Ascend CANN', value: 'npu' }, + { label: 'Moore Threads MUSA', value: 'musa' }, + { label: 'Hygon DCU', value: 'dcu' }, { label: 'CPU', value: 'cpu' } ];