Update the way toggle GPU selection

form.js

@@ -27,9 +27,10 @@ function set_partition_change_handler() {
 }
 
 function toggle_gpu_visibility(event) {
-  const show = /pascal|ampere/.test(event.target.value);
-  const selector = '#batch_connect_session_context_bc_num_gpus';
-  toggle_visibilty_of_form_group(selector, show);
+  const partition = event.target.value;
+  const gpu_selector = '#batch_connect_session_context_bc_num_gpus';
+
+  toggle_visibilty_of_form_group(gpu_selector, table[partition]["max_gpu"] != -1);
 }
 
 function toggle_visibilty_of_form_group(form_id, show) {