skypilot-org / skypilot

SkyPilot: Run LLMs, AI, and Batch jobs on any cloud. Get maximum savings, highest GPU availability, and managed execution—all with a simple interface.
https://skypilot.readthedocs.io
Apache License 2.0
6.2k stars 426 forks source link

Runpod uses extremely old base image #3542

Open baxelrod opened 1 month ago

baxelrod commented 1 month ago

I'm excited to use skypilot across multiple clouds, but I've noticed that it uses a really old runpod base image that has a very outdated cuda version.

https://github.com/skypilot-org/skypilot/blob/8a0a34d36bc75cd1e337e11d14e5bf8aeb4144e4/sky/provision/runpod/utils.py#L105

Would you consider a pull request to update to a more recent version or removing the version tag? Please let me know if there is anything I can do to make this happen.

They are currently on version 0.6.2. Source.

concretevitamin commented 1 month ago

I wonder if it’s sufficient to just change that image version and test everything still works? Cc @justinmerrell