-
Notifications
You must be signed in to change notification settings - Fork 6
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #10 from joshanderson-kw/iqr_demo
Iqr web demo
- Loading branch information
Showing
32 changed files
with
1,419 additions
and
299 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,10 @@ | ||
| SMQTK_REGISTRY=kitware/smqtk | ||
|
|
||
| # Dependency Image/Source versions to be used | ||
| CUDA_DEVEL_IMAGE_TAG=9.2-cudnn7-devel-ubuntu18.04 | ||
| CUDA_RUNTIME_IMAGE_TAG=9.2-cudnn7-runtime-ubuntu18.04 | ||
|
|
||
| FAISS_VERSION=1.6.2 | ||
|
|
||
| # OUR Images | ||
| SMQTK_TAG=latest |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,105 @@ | ||
| version: "3.4" | ||
|
|
||
| services: | ||
|
|
||
| # | ||
| # TPL: Caffe 1.0 | ||
| # | ||
| tpl-caffe-base: | ||
| # Base stage for common use by CPU and GPU build variants. | ||
| image: ${SMQTK_REGISTRY}/tpl-caffe:1.0-base | ||
| build: | ||
| context: tpl/caffe1 | ||
| target: base | ||
| args: | ||
| CUDA_DEVEL_IMAGE_TAG: # using value from .env file. | ||
|
|
||
| tpl-caffe-cpu: | ||
| # CPU-Only build variant | ||
| image: ${SMQTK_REGISTRY}/tpl-caffe:1.0-cpu | ||
| depends_on: | ||
| - tpl-caffe-base | ||
| build: | ||
| context: tpl/caffe1 | ||
| target: caffe_cpu | ||
| args: | ||
| CUDA_DEVEL_IMAGE_TAG: # using value from .env file. | ||
|
|
||
| tpl-caffe-gpu: | ||
| # GPU build variant | ||
| image: ${SMQTK_REGISTRY}/tpl-caffe:1.0-cuda${CUDA_DEVEL_IMAGE_TAG} | ||
| depends_on: | ||
| - tpl-caffe-base | ||
| build: | ||
| context: tpl/caffe1 | ||
| target: caffe_gpu | ||
| args: | ||
| CUDA_DEVEL_IMAGE_TAG: # using value from .env file. | ||
| # One or more CUDA architecture binary versions to compile for. | ||
| CUDA_ARCH_BIN_LIST: 50 52 60 61 62 70 72 | ||
|
|
||
| # | ||
| # TPL: FAISS | ||
| # | ||
| tpl-faiss-gpu: | ||
| image: ${SMQTK_REGISTRY}/tpl-faiss:${FAISS_VERSION}-cuda${CUDA_DEVEL_IMAGE_TAG} | ||
| build: | ||
| context: tpl/faiss | ||
| dockerfile: gpu/Dockerfile | ||
| args: | ||
| FAISS_VERSION: # using value from .env file. | ||
| CUDA_DEVEL_IMAGE_TAG: # using value from .env file. | ||
| # One or more CUDA architectures to compile for. | ||
| # Using ">-" for multi-line string concatenation with no new-lines. | ||
| # Using ">-" for multi-line string concatenation with no new-lines. | ||
| # - arch upper bound defined by CUDA version (e.g. 9.2 cannot support | ||
| # compute_75) | ||
| # - https://docs.nvidia.com/cuda/turing-compatibility-guide/index.html#building-turing-compatible-apps-using-cuda-9- | ||
| # - In order to support a NEWER gpu architecture than considered here, | ||
| # the final line here should specify a PTX version, e.g. | ||
| # `-gencode=arch=compute_XY,code=compute_XY` (note the `code=...`), | ||
| # in order to provide compatibility with future architectures. | ||
| CUDA_ARCH: >- | ||
| -gencode=arch=compute_50,code=sm_50 | ||
| -gencode=arch=compute_52,code=sm_52 | ||
| -gencode=arch=compute_60,code=sm_60 | ||
| -gencode=arch=compute_61,code=sm_61 | ||
| -gencode=arch=compute_62,code=sm_62 | ||
| -gencode=arch=compute_70,code=sm_70 | ||
| -gencode=arch=compute_72,code=sm_72 | ||
| -gencode=arch=compute_72,code=compute_72 | ||
| # | ||
| # IQR Playground Demo Image CPU/GPU | ||
| # | ||
| iqr-playground-cpu: | ||
| image: ${SMQTK_REGISTRY}/iqr_playground:${SMQTK_TAG}-cpu | ||
| depends_on: | ||
| - tpl-caffe-cpu | ||
| - tpl-faiss-gpu | ||
| build: | ||
| # Repo root | ||
| context: .. | ||
| dockerfile: docker/smqtk_iqr_playground/Dockerfile | ||
| target: smqtk-cpu | ||
| args: | ||
| CUDA_RUNTIME_IMAGE_TAG: # using value from .env file. | ||
| CAFFE_COPY_FROM_IMAGE: ${SMQTK_REGISTRY}/tpl-caffe:1.0-cpu | ||
| FAISS_COPY_FROM_IMAGE: ${SMQTK_REGISTRY}/tpl-faiss:${FAISS_VERSION}-cuda${CUDA_DEVEL_IMAGE_TAG} | ||
| FAISS_VERSION: # using value from .env file. | ||
|
|
||
| iqr-playground-gpu: | ||
| image: ${SMQTK_REGISTRY}/iqr_playground:${SMQTK_TAG}-cuda${CUDA_RUNTIME_IMAGE_TAG} | ||
| depends_on: | ||
| - tpl-caffe-gpu | ||
| - tpl-faiss-gpu | ||
| build: | ||
| # Repo root | ||
| context: .. | ||
| dockerfile: docker/smqtk_iqr_playground/Dockerfile | ||
| target: smqtk-gpu | ||
| args: | ||
| CUDA_RUNTIME_IMAGE_TAG: # using value from .env file. | ||
| CAFFE_COPY_FROM_IMAGE: ${SMQTK_REGISTRY}/tpl-caffe:1.0-cuda${CUDA_DEVEL_IMAGE_TAG} | ||
| FAISS_COPY_FROM_IMAGE: ${SMQTK_REGISTRY}/tpl-faiss:${FAISS_VERSION}-cuda${CUDA_DEVEL_IMAGE_TAG} | ||
| FAISS_VERSION: # using value from .env file. |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.