diff --git a/install.js b/install.js
index dd20c8d..b13226f 100644
--- a/install.js
+++ b/install.js
@@ -18,7 +18,7 @@ function install(kernel)
 	}
 	if ((platform === 'linux' || platform === 'win32') && gpu === 'nvidia')
 	{
-		return 'python install.py --onnxruntime cuda';
+		return 'python install.py --onnxruntime cuda@12';
 	}
 	return 'python install.py --onnxruntime default';
 }