mirror of
https://github.com/kodxana/madiator-docker-runpod.git
synced 2024-11-10 05:25:18 +01:00
Added CUDA 12.4 Pytorch Base
This commit is contained in:
parent
a833efc5d8
commit
37c08d4b55
1 changed files with 16 additions and 1 deletions
|
@ -1,5 +1,5 @@
|
||||||
group "default" {
|
group "default" {
|
||||||
targets = ["py310-cuda121", "py310-cuda118"]
|
targets = ["py310-cuda121", "py310-cuda118", "py310-cuda124"]
|
||||||
}
|
}
|
||||||
|
|
||||||
target "py310-cuda121" {
|
target "py310-cuda121" {
|
||||||
|
@ -31,3 +31,18 @@ target "py310-cuda118" {
|
||||||
}
|
}
|
||||||
tags = ["madiator2011/better-pytorch:cuda11.8"]
|
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"]
|
||||||
|
}
|
||||||
|
|
Loading…
Reference in a new issue