diff --git a/src/pages/resources/components/container-install.tsx b/src/pages/resources/components/container-install.tsx index fdc2e2af..2351313a 100644 --- a/src/pages/resources/components/container-install.tsx +++ b/src/pages/resources/components/container-install.tsx @@ -33,21 +33,15 @@ const AddWorker: React.FC = (props) => { let commandCode = addWorkerGuide[activeKey]; - if (npuKey === 'npu310p') { + if (npuKey === 'npu310p' && activeKey === 'npu') { commandCode = addWorkerGuide[npuKey]; } - if (activeKey === 'cuda') { - return commandCode?.registerWorker({ - server: origin, - tag: version, - token: '${token}', - workerip: '${workerip}' - }); - } + const tag = activeKey === 'cuda' ? version : `${version}-${activeKey}`; + return commandCode?.registerWorker({ server: origin, - tag: `${version}-${activeKey}`, + tag: tag, token: '${token}', workerip: '${workerip}' });