Python client for the Vast.ai cloud rent service. This package provides a Python client, that allows to list, create, destroy GPU instances programmaticaly, using Python.
There is an official Vast.ai CLI, however it can only be used through the command line.
pip install vastai-client
With the package you can list offers and run selected machines:
from vastai_client import VastClient
client = VastClient(api_key=<your_api_key>)
available_machines = client.search_offers(search_query='reliability > 0.98 num_gpus=1 gpu_name=RTX_3090', sort_order='dph-')
print(available_machines)
selected_machine = available_machines[0]
client.create_instance(id=selected_machine.id, image='pytorch/pytorch', ssh=True)
For more details, watch documentation.
This project was generated with wemake-python-package
. Current template version is: 9899cb192f754a566da703614227e6d63227b933. See what is updated since then.