67 lines
1.7 KiB
GLSL
Executable File
67 lines
1.7 KiB
GLSL
Executable File
#!/usr/bin/env -S v -n -w -gc none -cc tcc -d use_openssl -enable-globals run
|
|
|
|
import freeflowuniverse.herolib.clients.vastai
|
|
import json
|
|
import x.json2
|
|
|
|
// Create client with direct API key
|
|
// This uses VASTAI_API_KEY from environment
|
|
mut va := vastai.get()!
|
|
|
|
offers := va.search_offers()!
|
|
println('offers: ${offers}')
|
|
|
|
top_offers := va.get_top_offers(5)!
|
|
println('top offers: ${top_offers}')
|
|
|
|
create_instance_res := va.create_instance(
|
|
id: top_offers[0].id
|
|
config: vastai.CreateInstanceConfig{
|
|
image: 'pytorch/pytorch:2.5.1-cuda12.4-cudnn9-runtime'
|
|
disk: 10
|
|
}
|
|
)!
|
|
println('create instance res: ${create_instance_res}')
|
|
|
|
attach_sshkey_to_instance_res := va.attach_sshkey_to_instance(
|
|
id: 1
|
|
ssh_key: 'ssh-rsa AAAA...'
|
|
)!
|
|
println('attach sshkey to instance res: ${attach_sshkey_to_instance_res}')
|
|
|
|
stop_instance_res := va.stop_instance(
|
|
id: 1
|
|
state: 'stopped'
|
|
)!
|
|
println('stop instance res: ${stop_instance_res}')
|
|
|
|
destroy_instance_res := va.destroy_instance(
|
|
id: 1
|
|
)!
|
|
println('destroy instance res: ${destroy_instance_res}')
|
|
|
|
// For some reason this method returns an error from their server, 500 ERROR
|
|
// (request failed with code 500: {"error":"server_error","msg":"Something went wrong on the server"})
|
|
launch_instance_res := va.launch_instance(
|
|
// Required
|
|
num_gpus: 1
|
|
gpu_name: 'RTX_3090'
|
|
image: 'vastai/tensorflow'
|
|
disk: 10
|
|
region: 'us-west'
|
|
|
|
// Optional
|
|
env: 'user=7amada, home=/home/7amada'
|
|
)!
|
|
println('destroy instance res: ${launch_instance_res}')
|
|
|
|
start_instances_res := va.start_instances(
|
|
ids: [1, 2, 3]
|
|
)!
|
|
println('start instances res: ${start_instances_res}')
|
|
|
|
start_instance_res := va.start_instance(
|
|
id: 1
|
|
)!
|
|
println('start instance res: ${start_instance_res}')
|