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

Added CUDA 12.4 Pytorch Base

Browse source
This commit is contained in:
Madiator2011 2024-07-13 20:20:15 +02:00 committed by GitHub
parent a833efc5d8
commit 37c08d4b55
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 16 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
official-templates/better-pytorch/docker-bake.hcl
View file

@ -1,5 +1,5 @@
group "default" {
targets = ["py310-cuda121", "py310-cuda118"]
targets = ["py310-cuda121", "py310-cuda118", "py310-cuda124"]
}
target "py310-cuda121" {
@ -31,3 +31,18 @@ target "py310-cuda118" {
}
tags = ["madiator2011/better-pytorch:cuda11.8"]
}
target "py310-cuda124" {
contexts = {
scripts = "../../container-template"
proxy = "../../container-template/proxy"
logo = "../../container-template"
}
dockerfile = "Dockerfile"
args = {
BASE_IMAGE = "nvidia/cuda:12.4.1-cudnn-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.4"]
}