Barahlush / vastai_client

This package provides a Python client for the Vast.ai GPU rent service, that allows to list, create, destroy GPU instances programmaticaly, using Python.
MIT License
4 stars 5 forks source link
gpu hosting python rent vast vast-ai

vastai-client

Python Version wemake-python-styleguide

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.

Installation

pip install vastai-client

Example

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.

License

MIT

Credits

This project was generated with wemake-python-package. Current template version is: 9899cb192f754a566da703614227e6d63227b933. See what is updated since then.