Update docker-bake.hcl

This commit is contained in:
Madiator2011 2024-08-06 13:14:53 +02:00 committed by GitHub
parent f0457bdff4
commit 29de5b6648
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -23,6 +23,7 @@ group "rocm" {
] ]
} }
# CUDA Targets
target "py311-cuda121" { target "py311-cuda121" {
contexts = { contexts = {
default = "../../container-template" default = "../../container-template"
@ -65,6 +66,7 @@ target "py311-cuda124" {
tags = ["madiator2011/better-base:cuda12.4"] tags = ["madiator2011/better-base:cuda12.4"]
} }
# ROCm Targets
target "py310-rocm56" { target "py310-rocm56" {
contexts = { contexts = {
default = "../../container-template" default = "../../container-template"
@ -73,7 +75,7 @@ target "py310-rocm56" {
} }
dockerfile = "Dockerfile" dockerfile = "Dockerfile"
args = { args = {
BASE_IMAGE = "rocm/pytorch:rocm5.6_ubuntu20.04_py3.10_pytorch_2.0.1", BASE_IMAGE = "rocm/pytorch:rocm5.6_ubuntu20.04_py3.8_pytorch_2.0.1",
PYTHON_VERSION = "3.10" PYTHON_VERSION = "3.10"
} }
tags = ["madiator2011/better-base:rocm5.6"] tags = ["madiator2011/better-base:rocm5.6"]
@ -115,7 +117,7 @@ target "py310-rocm60" {
} }
dockerfile = "Dockerfile" dockerfile = "Dockerfile"
args = { args = {
BASE_IMAGE = "rocm/pytorch:rocm6.0_ubuntu20.04_py3.10_pytorch_2.1.1", BASE_IMAGE = "rocm/pytorch:rocm6.0_ubuntu20.04_py3.9_pytorch_2.1.1",
PYTHON_VERSION = "3.10" PYTHON_VERSION = "3.10"
} }
tags = ["madiator2011/better-base:rocm6.0"] tags = ["madiator2011/better-base:rocm6.0"]