2022-06-14 16:29:12 +08:00
|
|
|
DOCKER_REGISTRY = registry.cn-shanghai.aliyuncs.com
|
|
|
|
|
DOCKER_ORG = modelscope
|
|
|
|
|
DOCKER_IMAGE = modelscope
|
|
|
|
|
DOCKER_FULL_NAME = $(DOCKER_REGISTRY)/$(DOCKER_ORG)/$(DOCKER_IMAGE)
|
|
|
|
|
|
|
|
|
|
# CUDA_VERSION = 11.3
|
|
|
|
|
# CUDNN_VERSION = 8
|
|
|
|
|
BASE_RUNTIME = reg.docker.alibaba-inc.com/pai-dlc/pytorch-training:1.10PAI-gpu-py36-cu113-ubuntu18.04
|
2022-06-16 11:15:09 +08:00
|
|
|
# BASE_DEVEL = reg.docker.alibaba-inc.com/pai-dlc/pytorch-training:1.10PAI-gpu-py36-cu113-ubuntu18.04
|
|
|
|
|
BASE_DEVEL = pytorch/pytorch:1.10.0-cuda11.3-cudnn8-devel
|
2022-06-14 16:29:12 +08:00
|
|
|
|
|
|
|
|
|
|
|
|
|
MODELSCOPE_VERSION = $(shell git describe --tags --always)
|
|
|
|
|
|
|
|
|
|
# Can be either official / dev
|
|
|
|
|
BUILD_TYPE = dev
|
|
|
|
|
BUILD_PROGRESS = auto
|
|
|
|
|
BUILD_ARGS = --build-arg BASE_IMAGE=$(BASE_IMAGE)
|
|
|
|
|
|
|
|
|
|
EXTRA_DOCKER_BUILD_FLAGS ?= --network=host
|
|
|
|
|
# DOCKER_BUILD = DOCKER_BUILDKIT=1 \
|
|
|
|
|
# docker build \
|
|
|
|
|
# --progress=$(BUILD_PROGRESS) \
|
|
|
|
|
# $(EXTRA_DOCKER_BUILD_FLAGS) \
|
|
|
|
|
# --target $(BUILD_TYPE) \
|
|
|
|
|
# -t $(DOCKER_FULL_NAME):$(DOCKER_TAG) \
|
|
|
|
|
# $(BUILD_ARGS) \
|
|
|
|
|
# -f docker/pytorch.dockerfile .
|
|
|
|
|
DOCKER_BUILD = DOCKER_BUILDKIT=1 \
|
|
|
|
|
docker build \
|
|
|
|
|
$(EXTRA_DOCKER_BUILD_FLAGS) \
|
|
|
|
|
-t $(DOCKER_FULL_NAME):$(DOCKER_TAG) \
|
|
|
|
|
$(BUILD_ARGS) \
|
|
|
|
|
-f docker/pytorch.dockerfile .
|
|
|
|
|
DOCKER_PUSH = docker push $(DOCKER_FULL_NAME):$(DOCKER_TAG)
|
|
|
|
|
|
|
|
|
|
.PHONY: all
|
|
|
|
|
all: devel-image
|
|
|
|
|
|
|
|
|
|
.PHONY: devel-image
|
|
|
|
|
devel-image: BASE_IMAGE := $(BASE_DEVEL)
|
|
|
|
|
devel-image: DOCKER_TAG := $(MODELSCOPE_VERSION)-devel
|
|
|
|
|
devel-image:
|
|
|
|
|
$(DOCKER_BUILD)
|
|
|
|
|
|
|
|
|
|
.PHONY: devel-push
|
|
|
|
|
devel-push: BASE_IMAGE := $(BASE_DEVEL)
|
|
|
|
|
devel-push: DOCKER_TAG := $(MODELSCOPE_VERSION)-devel
|
|
|
|
|
devel-push:
|
|
|
|
|
$(DOCKER_PUSH)
|
|
|
|
|
|
|
|
|
|
.PHONY: runtime-image
|
|
|
|
|
runtime-image: BASE_IMAGE := $(BASE_RUNTIME)
|
|
|
|
|
runtime-image: DOCKER_TAG := $(MODELSCOPE_VERSION)-runtime
|
|
|
|
|
runtime-image:
|
|
|
|
|
$(DOCKER_BUILD)
|
|
|
|
|
docker tag $(DOCKER_FULL_NAME):$(DOCKER_TAG) $(DOCKER_FULL_NAME):latest
|
|
|
|
|
|
|
|
|
|
.PHONY: runtime-push
|
|
|
|
|
runtime-push: BASE_IMAGE := $(BASE_RUNTIME)
|
|
|
|
|
runtime-push: DOCKER_TAG := $(MODELSCOPE_VERSION)-runtime
|
|
|
|
|
runtime-push:
|
|
|
|
|
$(DOCKER_PUSH)
|
|
|
|
|
|
|
|
|
|
.PHONY: clean
|
|
|
|
|
clean:
|
|
|
|
|
-docker rmi -f $(shell docker images -q $(DOCKER_FULL_NAME))
|
