Madiator2011/madiator-docker-runpod
1
0
Fork 0
mirror of https://github.com/kodxana/madiator-docker-runpod.git synced 2024-11-10 05:25:18 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update docker-bake.hcl

Browse source
This commit is contained in:
Madiator2011 2024-08-06 13:02:18 +02:00 committed by GitHub
parent 3cedf754d3
commit f8f9d548ad
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 49 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

50
official-templates/better-base/docker-bake.hcl
View file

@ -3,8 +3,13 @@ group "default" {
"py311-cuda121", "py311-cuda121",
"py311-cuda118", "py311-cuda118",
"py311-cuda124", "py311-cuda124",
"py310-rocm56", "py38-rocm56",
"py310-rocm57", "py310-rocm57",
"py310-rocm602",
"py39-rocm60",
"py39-rocm61",
"py310-rocm61",
"py39-rocm612",
"py310-rocm612" "py310-rocm612"
] ]
} }
@ -51,6 +56,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 = {
scripts = "../../container-template" scripts = "../../container-template"
@ -79,6 +85,48 @@ target "py310-rocm57" {
tags = ["madiator2011/better-base:rocm5.7"] tags = ["madiator2011/better-base:rocm5.7"]
} }
target "py310-rocm602" {
contexts = {
scripts = "../../container-template"
proxy = "../../container-template/proxy"
logo = "../../container-template"
}
dockerfile = "Dockerfile"
args = {
BASE_IMAGE = "rocm/pytorch:rocm6.0.2_ubuntu22.04_py3.10_pytorch_2.1.2",
PYTHON_VERSION = "3.10"
}
tags = ["madiator2011/better-base:rocm6.0.2"]
}
target "py310-rocm60" {
contexts = {
scripts = "../../container-template"
proxy = "../../container-template/proxy"
logo = "../../container-template"
}
dockerfile = "Dockerfile"
args = {
BASE_IMAGE = "rocm/pytorch:rocm6.0_ubuntu20.04_py3.10_pytorch_2.1.1",
PYTHON_VERSION = "3.10"
}
tags = ["madiator2011/better-base:rocm6.0"]
}
target "py310-rocm61" {
contexts = {
scripts = "../../container-template"
proxy = "../../container-template/proxy"
logo = "../../container-template"
}
dockerfile = "Dockerfile"
args = {
BASE_IMAGE = "rocm/pytorch:rocm6.1_ubuntu22.04_py3.10_pytorch_2.1.2",
PYTHON_VERSION = "3.10"
}
tags = ["madiator2011/better-base:rocm6.1"]
}
target "py310-rocm612" { target "py310-rocm612" {
contexts = { contexts = {
scripts = "../../container-template" scripts = "../../container-template"