slurm-cluster-gpu.yaml

# Copyright 2017 SchedMD LLC.
# Modified for use with the Slurm Resource Manager.
#
# Copyright 2015 Google Inc. All rights reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

# [START cluster_yaml]
imports:
- path: slurm.jinja

resources:
- name: slurm-cluster
  type: slurm.jinja
  properties:
    cluster_name            : datta-gpu
    static_node_count       : 0
    max_node_count          : 50

    zone                    : us-east1-c
    region                  : us-east1
    cidr                    : 10.10.0.0/16

    controller_machine_type : n1-standard-8
    compute_machine_type    : n1-standard-16
    login_machine_type      : n1-standard-2
    login_node_count        : 1

    # Optional compute configuration fields
    #cpu_platform               : Intel Skylake
    preemptible_bursting       : True
    #external_compute_ips       : False
    private_google_access      : True

    compute_disk_type          : pd-standard
    compute_disk_size_gb       : 50
    controller_disk_type       : pd-standard
    controller_disk_size_gb    : 2000
    login_disk_type            : pd-standard
    login_disk_size_gb         : 10
    #nfs_apps_server            :
    #nfs_home_server            :
    #controller_secondary_disk          : True
    #controller_secondary_disk_type     : pd-standard
    #controller_secondary_disk_size_gb  : 2000

    # Optional GPU configuration fields
    gpu_type                   : nvidia-tesla-p100
    gpu_count                  : 1

    slurm_version           : 18.08.4
    default_account         : default
    default_users           : dattalab, jmarkow, ralph, wgillis, diegoaldarondo
#  [END cluster_yaml]