scripts/packer/build-cuda-image.pkr.hcl

Rama master · réplica de lectura del repositorio original.

build {
  # TODO: transition to this generic template from legacy per-backend JSON templates

  provisioner "shell" {
    inline = ["cloud-init status --long --wait"]
  }

  provisioner "shell" {
    scripts = ["provisioners/kernel/apt-upgrade.sh", "provisioners/kernel/apt-daily.sh", "provisioners/kernel/apt-packages.sh", "provisioners/kernel/kernel-tuning.sh"]
  }

  provisioner "file" {
    destination = "/tmp/install-docker.sh"
    source      = "provisioners/install-docker.sh"
  }

  provisioner "file" {
    destination = "/tmp/run-docker"
    source      = "provisioners/run-docker"
  }

  provisioner "shell" {
    inline = ["cd /tmp", "chmod +x install-docker.sh", "./install-docker.sh --version ${local.docker_version}"]
  }

  provisioner "shell" {
    environment_vars = ["CUDA_DRIVERS_VERSION=${local.cuda_drivers_version}"]
    script           = "provisioners/cuda.sh"
  }

  provisioner "shell" {
    script = "provisioners/install-nvidia-container-toolkit.sh"
  }
}