notebooks/components/example-notebook-servers/jupyter-pytorch-full/Makefile

21 lines
796 B
Makefile

TAG ?= $(shell git describe --tags --always --dirty)
REGISTRY ?= kubeflownotebookswg
docker-build-jupyter-pytorch:
make docker-build-cpu -C ../jupyter-pytorch TAG=${TAG}
docker-build-jupyter-pytorch-cuda:
make docker-build-cuda -C ../jupyter-pytorch TAG=${TAG}
docker-build-cpu: docker-build-jupyter-pytorch
docker build -t ${REGISTRY}/jupyter-pytorch-full:${TAG} --build-arg BASE_IMG=${REGISTRY}/jupyter-pytorch:${TAG} -f cpu.Dockerfile .
docker-build-cuda: docker-build-jupyter-pytorch-cuda
docker build -t ${REGISTRY}/jupyter-pytorch-cuda-full:${TAG} --build-arg BASE_IMG=${REGISTRY}/jupyter-pytorch-cuda:${TAG} -f cuda.Dockerfile .
docker-push-cpu:
docker push ${REGISTRY}/jupyter-pytorch-full:${TAG}
docker-push-cuda:
docker push ${REGISTRY}/jupyter-pytorch-cuda-full:${TAG}