diff --git a/frontend/src/utils/index.tsx b/frontend/src/utils/index.tsx
index fc882bb..62b8d2f 100644
--- a/frontend/src/utils/index.tsx
+++ b/frontend/src/utils/index.tsx
@@ -298,7 +298,7 @@ export function toastWithButton(text: string, buttonText: string, onClickButton:
 }
 
 export function getSupportedCustomCudaFile() {
-  if ([' 10', ' 16', ' 20', ' 30', 'P40'].some(v => commonStore.status.device_name.includes(v)))
+  if ([' 10', ' 16', ' 20', ' 30', 'P40', 'P104', 'P106'].some(v => commonStore.status.device_name.includes(v)))
     return './backend-python/wkv_cuda_utils/wkv_cuda10_30.pyd';
   else if ([' 40'].some(v => commonStore.status.device_name.includes(v)))
     return './backend-python/wkv_cuda_utils/wkv_cuda40.pyd';