Skip to content

Commit 7719d73

Browse files
Add cuda-toolkit to a3ultra-jbvms blueprint
1 parent 7bd271d commit 7719d73

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

examples/machine-learning/a3-ultragpu-8g/a3ultra-vm.yaml

+8
Original file line numberDiff line numberDiff line change
@@ -105,6 +105,14 @@ deployment_groups:
105105
settings:
106106
configure_ssh_host_patterns:
107107
- $(vars.hostname_prefix)-*
108+
runners:
109+
- type: shell
110+
destination: install-cuda-toolkit.sh
111+
content: |
112+
#!/bin/bash
113+
set -e -o pipefail
114+
sudo add-nvidia-repositories -y
115+
sudo apt install -y cuda-toolkit-12-8
108116
109117
- id: a3ultra-vms
110118
source: modules/compute/vm-instance

0 commit comments

Comments
 (0)