| # from huggingface_hub import HfApi | |
| # api = HfApi() | |
| # job = api.create_job( | |
| # repo_id="uav-vts/training_sem", | |
| # docker_image="pytorch/pytorch:2.6.0-cuda12.4-cudnn9-devel", | |
| # command="python runner/train.py", | |
| # hardware="a100", | |
| # # token="<TOKEN>" | |
| # ) | |
| # print("Job created:", job.job_id) | |
| from huggingface_hub import run_job | |
| image = "pytorch/pytorch:2.6.0-cuda12.4-cudnn9-devel" | |
| command = [ | |
| "python", | |
| "-c", | |
| "import torch; print(f'This code ran with the following GPU: {torch.cuda.get_device_name()}')" | |
| ] | |
| job = run_job( | |
| image=image, | |
| command=command, | |
| flavor="a100-large", | |
| namespace="uav-vts" | |
| ) | |
| print("Job submitted!") | |
| print("Track it here:", job.url) | |