mirror of
https://github.com/kodxana/madiator-docker-runpod.git
synced 2024-11-10 05:25:18 +01:00
Update docker-bake.hcl
This commit is contained in:
parent
f0457bdff4
commit
29de5b6648
1 changed files with 4 additions and 2 deletions
|
@ -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"]
|
||||||
|
|
Loading…
Reference in a new issue