From 37c08d4b553b979f2cf0e5af84fc4cf633b4e231 Mon Sep 17 00:00:00 2001 From: Madiator2011 Date: Sat, 13 Jul 2024 20:20:15 +0200 Subject: [PATCH] Added CUDA 12.4 Pytorch Base --- .../better-pytorch/docker-bake.hcl | 17 ++++++++++++++++- 1 file changed, 16 insertions(+), 1 deletion(-) 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"] +}