Create docker-bake.hcl

This commit is contained in:
Madiator2011 2024-06-16 13:39:23 +02:00 committed by GitHub
parent 24a8925c93
commit c6f51873ca
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View 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"]
}