cu118-torch212 | cu118-torch222 | cu121-torch221 | |
---|---|---|---|
Ubuntu | 22.04 LTS | 22.04 LTS | 22.04 LTS |
CUDA | 11.8 | 11.8 | 12.1 |
Python | 3.10.12 | 3.10.12 | 3.10.12 |
Torch | 2.1.2 | 2.2.2 | 2.2.1 |
xformers | 0.0.23.post1 | 0.0.25.post1 | None |
Jupyter Lab | |||
runpodctl | |||
OhMyRunPod | |||
RunPod File Uploader | |||
croc | |||
rclone |
Note
You will need to edit the docker-bake.hcl
file and update REGISTRY_USER
,
and RELEASE
. You can obviously edit the other values too, but these
are the most important ones.
# Clone the repo
git clone https://github.com/ashleykleynhans/runpod-base-images.git
# Log in to Docker Hub
docker login
# Build the image, tag the image, and push the image to Docker Hub
docker buildx bake -f docker-bake.hcl --push
# Build a specific target for a different user, registry and release version
REGISTRY=ghcr.io REGISTRY_USER=my_gh_user RELEASE=x.y.z docker buildx bake \
-f docker-bake.hcl --push cu118-torch212
Pull requests and issues on GitHub are welcome. Bug fixes and new features are encouraged.