Lines Matching +full:build +full:- +full:docker +full:- +full:cuda
1 DOCKER_REGISTRY ?= docker.io
2 DOCKER_ORG ?= $(shell docker info 2>/dev/null | sed '/Username:/!d;s/.* //')
7 $(warning WARNING: No docker user found using results from whoami)
15 BASE_DEVEL = nvidia/cuda:$(CUDA_VERSION)-devel-ubuntu22.04
30 PYTORCH_VERSION ?= $(shell git describe --tags --always --match "v[1-9]*.*")
36 BUILD_ARGS = --build-arg BASE_IMAGE=$(BASE_IMAGE) \
37 --build-arg PYTHON_VERSION=$(PYTHON_VERSION) \
38 --build-arg CUDA_VERSION=$(CUDA_VERSION) \
39 --build-arg CUDA_PATH=$(CUDA_PATH) \
40 --build-arg PYTORCH_VERSION=$(PYTORCH_VERSION) \
41 --build-arg INSTALL_CHANNEL=$(INSTALL_CHANNEL) \
42 --build-arg TRITON_VERSION=$(TRITON_VERSION) \
43 --build-arg CMAKE_VARS="$(CMAKE_VARS)"
46 BUILD ?= build
55 BUILD = buildx build
57 PLATFORMS_FLAG = --platform="$(BUILD_PLATFORMS)"
61 PUSH_FLAG = --push
65 DOCKER_BUILD = docker $(BUILD) \
66 --progress=$(BUILD_PROGRESS) \
70 --target $(BUILD_TYPE) \
71 -t $(DOCKER_FULL_NAME):$(DOCKER_TAG) \
73 DOCKER_PUSH = docker push $(DOCKER_FULL_NAME):$(DOCKER_TAG)
76 all: devel-image
78 .PHONY: devel-image
79 devel-image: BASE_IMAGE := $(BASE_DEVEL)
80 devel-image: DOCKER_TAG := $(PYTORCH_VERSION)-cuda$(CUDA_VERSION_SHORT)-cudnn$(CUDNN_VERSION)-devel
81 devel-image:
84 .PHONY: devel-push
85 devel-push: BASE_IMAGE := $(BASE_DEVEL)
86 devel-push: DOCKER_TAG := $(PYTORCH_VERSION)-cuda$(CUDA_VERSION_SHORT)-cudnn$(CUDNN_VERSION)-devel
87 devel-push:
92 .PHONY: runtime-image
93 runtime-image: BASE_IMAGE := $(BASE_RUNTIME)
94 runtime-image: DOCKER_TAG := $(PYTORCH_VERSION)-runtime
95 runtime-image:
98 .PHONY: runtime-push
99 runtime-push: BASE_IMAGE := $(BASE_RUNTIME)
100 runtime-push: DOCKER_TAG := $(PYTORCH_VERSION)-runtime
101 runtime-push:
106 .PHONY: runtime-image
107 runtime-image: BASE_IMAGE := $(BASE_RUNTIME)
108 runtime-image: DOCKER_TAG := $(PYTORCH_VERSION)-cuda$(CUDA_VERSION_SHORT)-cudnn$(CUDNN_VERSION)-run…
109 runtime-image:
112 .PHONY: runtime-push
113 runtime-push: BASE_IMAGE := $(BASE_RUNTIME)
114 runtime-push: DOCKER_TAG := $(PYTORCH_VERSION)-cuda$(CUDA_VERSION_SHORT)-cudnn$(CUDNN_VERSION)-runt…
115 runtime-push:
122 -docker rmi -f $(shell docker images -q $(DOCKER_FULL_NAME))