diff --git a/src/pages/resources/config/gpu-driver.ts b/src/pages/resources/config/gpu-driver.ts index 75e101a1..531fa508 100644 --- a/src/pages/resources/config/gpu-driver.ts +++ b/src/pages/resources/config/gpu-driver.ts @@ -129,7 +129,7 @@ export const dockerEnvCommandMap = { [GPUDriverMap.HYGON]: generateHygonDockerEnvCommand( GPUsConfigs[GPUDriverMap.HYGON] ), - [GPUDriverMap.ILUVATAR]: generateHygonDockerEnvCommand( + [GPUDriverMap.ILUVATAR]: generateNvidiaDockerEnvCommand( GPUsConfigs[GPUDriverMap.ILUVATAR] ), [GPUDriverMap.CAMBRICON]: generateHygonDockerEnvCommand(