mirror of
https://github.com/kodxana/madiator-docker-runpod.git
synced 2024-11-10 05:25:18 +01:00
Create docker-bake.hcl
This commit is contained in:
parent
24a8925c93
commit
c6f51873ca
1 changed files with 33 additions and 0 deletions
33
official-templates/better-pytorch/docker-bake.hcl
Normal file
33
official-templates/better-pytorch/docker-bake.hcl
Normal file
|
@ -0,0 +1,33 @@
|
|||
group "default" {
|
||||
targets = ["py310-cuda121", "py310-cuda118"]
|
||||
}
|
||||
|
||||
target "py310-cuda121" {
|
||||
contexts = {
|
||||
scripts = "../../container-template"
|
||||
proxy = "../../container-template/proxy"
|
||||
logo = "../../container-template"
|
||||
}
|
||||
dockerfile = "Dockerfile"
|
||||
args = {
|
||||
BASE_IMAGE = "nvidia/cuda:12.1.0-cudnn8-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.1"]
|
||||
}
|
||||
|
||||
target "py310-cuda118" {
|
||||
contexts = {
|
||||
scripts = "../../container-template"
|
||||
proxy = "../../container-template/proxy"
|
||||
logo = "../../container-template"
|
||||
}
|
||||
dockerfile = "Dockerfile"
|
||||
args = {
|
||||
BASE_IMAGE = "nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04",
|
||||
TORCH = "torch==2.3.1+cu118 -f https://download.pytorch.org/whl/torch_stable.html",
|
||||
PYTHON_VERSION = "3.10"
|
||||
}
|
||||
tags = ["madiator2011/better-pytorch:cuda11.8"]
|
||||
}
|
Loading…
Reference in a new issue