Merge pull request #6 from crusoecloud/update-docs

Update IB Example
crusoecloud · Aug 18, 2023 · 69dd909 · 69dd909
2 parents 92e5fc2 + 1bb0e03
commit 69dd909
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/examples/infiniband/main.tf b/examples/infiniband/main.tf
@@ -28,10 +28,12 @@ resource "crusoe_ib_partition" "my_partition" {
 
 # create two VMs, both in the same Infiniband partition
 resource "crusoe_compute_instance" "my_vm1" {
-  count = 2
+  count = 8
 
   name = "ib-vm-${count.index}"
-  type = "h100-80gb-hgx.8x" # IB enabled VM type
+  type = "a100-80gb-sxm-ib.8x" # IB enabled VM type
+  location = "us-east1-a" # IB currently only supported at us-east1-a
+  image = "ubuntu20.04-nvidia-sxm-docker:ib-nccl2.18.3" # IB image
   ib_partition_id = crusoe_ib_partition.my_partition.id
   ssh_key = local.my_ssh_key