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

added ROCm images + update to py3.11

Browse source
This commit is contained in:
Madiator2011 2024-08-06 12:52:39 +02:00 committed by GitHub
parent 01ba79b48a
commit f77a1f07ee
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 56 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -1,8 +1,15 @@
group "default" { group "default" {
targets = ["py310-cuda121", "py310-cuda118", "py310-cuda124"] targets = [
"py311-cuda121",
"py311-cuda118",
"py311-cuda124",
"py311-rocm56",
"py311-rocm57",
"py311-rocm612"
]
} }
target "py310-cuda121" { target "py311-cuda121" {
contexts = { contexts = {
scripts = "../../container-template" scripts = "../../container-template"
proxy = "../../container-template/proxy" proxy = "../../container-template/proxy"
@ -11,12 +18,12 @@ target "py310-cuda121" {
dockerfile = "Dockerfile" dockerfile = "Dockerfile"
args = { args = {
BASE_IMAGE = "nvidia/cuda:12.1.0-cudnn8-devel-ubuntu22.04", BASE_IMAGE = "nvidia/cuda:12.1.0-cudnn8-devel-ubuntu22.04",
PYTHON_VERSION = "3.10" PYTHON_VERSION = "3.11"
} }
tags = ["madiator2011/better-base:cuda12.1"] tags = ["madiator2011/better-base:cuda12.1"]
} }
target "py310-cuda118" { target "py311-cuda118" {
contexts = { contexts = {
scripts = "../../container-template" scripts = "../../container-template"
proxy = "../../container-template/proxy" proxy = "../../container-template/proxy"
@ -25,12 +32,12 @@ target "py310-cuda118" {
dockerfile = "Dockerfile" dockerfile = "Dockerfile"
args = { args = {
BASE_IMAGE = "nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04", BASE_IMAGE = "nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04",
PYTHON_VERSION = "3.10" PYTHON_VERSION = "3.11"
} }
tags = ["madiator2011/better-base:cuda11.8"] tags = ["madiator2011/better-base:cuda11.8"]
} }
target "py310-cuda124" { target "py311-cuda124" {
contexts = { contexts = {
scripts = "../../container-template" scripts = "../../container-template"
proxy = "../../container-template/proxy" proxy = "../../container-template/proxy"
@ -39,7 +46,49 @@ target "py310-cuda124" {
dockerfile = "Dockerfile" dockerfile = "Dockerfile"
args = { args = {
BASE_IMAGE = "nvidia/cuda:12.4.1-cudnn-devel-ubuntu22.04", BASE_IMAGE = "nvidia/cuda:12.4.1-cudnn-devel-ubuntu22.04",
PYTHON_VERSION = "3.10" PYTHON_VERSION = "3.11"
} }
tags = ["madiator2011/better-base:cuda12.4"] tags = ["madiator2011/better-base:cuda12.4"]
} }
target "py311-rocm56" {
contexts = {
scripts = "../../container-template"
proxy = "../../container-template/proxy"
logo = "../../container-template"
}
dockerfile = "Dockerfile"
args = {
BASE_IMAGE = "rocm/pytorch:rocm5.6_ubuntu20.04_py3.11_pytorch_2.0.1",
PYTHON_VERSION = "3.11"
}
tags = ["madiator2011/better-base:rocm5.6"]
}
target "py311-rocm57" {
contexts = {
scripts = "../../container-template"
proxy = "../../container-template/proxy"
logo = "../../container-template"
}
dockerfile = "Dockerfile"
args = {
BASE_IMAGE = "rocm/pytorch:rocm5.7_ubuntu22.04_py3.11_pytorch_2.0.1",
PYTHON_VERSION = "3.11"
}
tags = ["madiator2011/better-base:rocm5.7"]
}
target "py311-rocm612" {
contexts = {
scripts = "../../container-template"
proxy = "../../container-template/proxy"
logo = "../../container-template"
}
dockerfile = "Dockerfile"
args = {
BASE_IMAGE = "rocm/pytorch:rocm6.1.2_ubuntu22.04_py3.11_pytorch_release-2.1.2",
PYTHON_VERSION = "3.11"
}
tags = ["madiator2011/better-base:rocm6.1.2"]
}