diff --git a/official-templates/better-pytorch/docker-bake.hcl b/official-templates/better-pytorch/docker-bake.hcl index dccb733..d4cb893 100644 --- a/official-templates/better-pytorch/docker-bake.hcl +++ b/official-templates/better-pytorch/docker-bake.hcl @@ -1,5 +1,5 @@ group "default" { - targets = ["py310-cuda121", "py310-cuda118"] + targets = ["py310-cuda121", "py310-cuda118", "py310-cuda124"] } target "py310-cuda121" { @@ -31,3 +31,18 @@ target "py310-cuda118" { } tags = ["madiator2011/better-pytorch:cuda11.8"] } + +target "py310-cuda124" { + contexts = { + scripts = "../../container-template" + proxy = "../../container-template/proxy" + logo = "../../container-template" + } + dockerfile = "Dockerfile" + args = { + BASE_IMAGE = "nvidia/cuda:12.4.1-cudnn-devel-ubuntu22.04", + TORCH = "torch==2.3.1+cu121 -f https://download.pytorch.org/whl/torch_stable.html", + PYTHON_VERSION = "3.10" + } + tags = ["madiator2011/better-pytorch:cuda12.4"] +}