diff --git a/.buildkite/pipeline.yml b/.buildkite/pipeline.yml index 730c7c16..52d99e3b 100644 --- a/.buildkite/pipeline.yml +++ b/.buildkite/pipeline.yml @@ -5,25 +5,25 @@ # steps: - - label: ":docker: CPU Tests (test-cpu-variant-tf-1_14_0-torch-1_3_0-py3_6)" + - label: ":docker: CPU Tests (test-cpu-variant-tf-2_5_0-torch-1_7_1-py3_9)" timeout_in_minutes: 60 agents: queue: public-gpu env: - NEUROPOD_TEST_FRAMEWORKS: tensorflow,torchscript,python + NEUROPOD_TEST_FRAMEWORKS: tensorflow command: build/ci/buildkite_build.sh plugins: - docker-compose#v3.7.0: - build: test-cpu-variant-tf-1_14_0-torch-1_3_0-py3_6 + build: test-cpu-variant-tf-2_5_0-torch-1_7_1-py3_9 config: docker-compose.test.yml image-repository: 027047743804.dkr.ecr.us-east-2.amazonaws.com/uber - cache-from: test-cpu-variant-tf-1_14_0-torch-1_3_0-py3_6:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-cpu-variant-tf-1_14_0-torch-1_3_0-py3_6 + cache-from: test-cpu-variant-tf-2_5_0-torch-1_7_1-py3_9:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-cpu-variant-tf-2_5_0-torch-1_7_1-py3_9 push-retries: 5 - docker-compose#v3.7.0: - push: test-cpu-variant-tf-1_14_0-torch-1_3_0-py3_6:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-cpu-variant-tf-1_14_0-torch-1_3_0-py3_6 + push: test-cpu-variant-tf-2_5_0-torch-1_7_1-py3_9:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-cpu-variant-tf-2_5_0-torch-1_7_1-py3_9 config: docker-compose.test.yml - docker-compose#v3.7.0: - run: test-cpu-variant-tf-1_14_0-torch-1_3_0-py3_6 + run: test-cpu-variant-tf-2_5_0-torch-1_7_1-py3_9 config: docker-compose.test.yml env: - NEUROPOD_CACHE_ACCESS_KEY @@ -53,303 +53,16 @@ steps: command: build/ci/buildkite_lint.sh plugins: - docker-compose#v3.7.0: - build: test-cpu-variant-tf-1_14_0-torch-1_3_0-py3_6 - config: docker-compose.test.yml - image-repository: 027047743804.dkr.ecr.us-east-2.amazonaws.com/uber - cache-from: test-cpu-variant-tf-1_14_0-torch-1_3_0-py3_6:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-cpu-variant-tf-1_14_0-torch-1_3_0-py3_6 - push-retries: 5 - - docker-compose#v3.7.0: - push: test-cpu-variant-tf-1_14_0-torch-1_3_0-py3_6:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-cpu-variant-tf-1_14_0-torch-1_3_0-py3_6 - config: docker-compose.test.yml - - docker-compose#v3.7.0: - run: test-cpu-variant-tf-1_14_0-torch-1_3_0-py3_6 - config: docker-compose.test.yml - env: - - NEUROPOD_CACHE_ACCESS_KEY - - NEUROPOD_CACHE_ACCESS_SECRET - - BUILDKITE - - BUILDKITE_BRANCH - - BUILDKITE_BUILD_NUMBER - - BUILDKITE_BUILD_URL - - BUILDKITE_COMMIT - - BUILDKITE_JOB_ID - - BUILDKITE_PROJECT_SLUG - - BUILDKITE_PULL_REQUEST - - BUILDKITE_TAG - - CI - - CODECOV_TOKEN - - GH_STATUS_TOKEN - - GH_UPLOAD_TOKEN - - NEUROPOD_TEST_FRAMEWORKS - - WEB_DEPLOY_KEY - retry: - automatic: true - - - label: ":docker: CPU Tests (test-cpu-variant-tf-1_15_0-torch-1_4_0-py3_7)" - timeout_in_minutes: 60 - agents: - queue: public-gpu - env: - NEUROPOD_TEST_FRAMEWORKS: tensorflow,torchscript,python - command: build/ci/buildkite_build.sh - plugins: - - docker-compose#v3.7.0: - build: test-cpu-variant-tf-1_15_0-torch-1_4_0-py3_7 - config: docker-compose.test.yml - image-repository: 027047743804.dkr.ecr.us-east-2.amazonaws.com/uber - cache-from: test-cpu-variant-tf-1_15_0-torch-1_4_0-py3_7:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-cpu-variant-tf-1_15_0-torch-1_4_0-py3_7 - push-retries: 5 - - docker-compose#v3.7.0: - push: test-cpu-variant-tf-1_15_0-torch-1_4_0-py3_7:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-cpu-variant-tf-1_15_0-torch-1_4_0-py3_7 - config: docker-compose.test.yml - - docker-compose#v3.7.0: - run: test-cpu-variant-tf-1_15_0-torch-1_4_0-py3_7 - config: docker-compose.test.yml - env: - - NEUROPOD_CACHE_ACCESS_KEY - - NEUROPOD_CACHE_ACCESS_SECRET - - BUILDKITE - - BUILDKITE_BRANCH - - BUILDKITE_BUILD_NUMBER - - BUILDKITE_BUILD_URL - - BUILDKITE_COMMIT - - BUILDKITE_JOB_ID - - BUILDKITE_PROJECT_SLUG - - BUILDKITE_PULL_REQUEST - - BUILDKITE_TAG - - CI - - CODECOV_TOKEN - - GH_STATUS_TOKEN - - GH_UPLOAD_TOKEN - - NEUROPOD_TEST_FRAMEWORKS - - WEB_DEPLOY_KEY - retry: - automatic: true - - - label: ":docker: CPU Tests (test-cpu-variant-tf-2_2_0-torch-1_5_0-py3_8)" - timeout_in_minutes: 60 - agents: - queue: public-gpu - env: - NEUROPOD_TEST_FRAMEWORKS: tensorflow,torchscript,python - command: build/ci/buildkite_build.sh - plugins: - - docker-compose#v3.7.0: - build: test-cpu-variant-tf-2_2_0-torch-1_5_0-py3_8 - config: docker-compose.test.yml - image-repository: 027047743804.dkr.ecr.us-east-2.amazonaws.com/uber - cache-from: test-cpu-variant-tf-2_2_0-torch-1_5_0-py3_8:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-cpu-variant-tf-2_2_0-torch-1_5_0-py3_8 - push-retries: 5 - - docker-compose#v3.7.0: - push: test-cpu-variant-tf-2_2_0-torch-1_5_0-py3_8:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-cpu-variant-tf-2_2_0-torch-1_5_0-py3_8 - config: docker-compose.test.yml - - docker-compose#v3.7.0: - run: test-cpu-variant-tf-2_2_0-torch-1_5_0-py3_8 - config: docker-compose.test.yml - env: - - NEUROPOD_CACHE_ACCESS_KEY - - NEUROPOD_CACHE_ACCESS_SECRET - - BUILDKITE - - BUILDKITE_BRANCH - - BUILDKITE_BUILD_NUMBER - - BUILDKITE_BUILD_URL - - BUILDKITE_COMMIT - - BUILDKITE_JOB_ID - - BUILDKITE_PROJECT_SLUG - - BUILDKITE_PULL_REQUEST - - BUILDKITE_TAG - - CI - - CODECOV_TOKEN - - GH_STATUS_TOKEN - - GH_UPLOAD_TOKEN - - NEUROPOD_TEST_FRAMEWORKS - - WEB_DEPLOY_KEY - retry: - automatic: true - - - label: ":docker: CPU Tests (test-cpu-variant-tf-2_2_0-torch-1_6_0-py3_8)" - timeout_in_minutes: 60 - agents: - queue: public-gpu - env: - NEUROPOD_TEST_FRAMEWORKS: torchscript,python - command: build/ci/buildkite_build.sh - plugins: - - docker-compose#v3.7.0: - build: test-cpu-variant-tf-2_2_0-torch-1_6_0-py3_8 - config: docker-compose.test.yml - image-repository: 027047743804.dkr.ecr.us-east-2.amazonaws.com/uber - cache-from: test-cpu-variant-tf-2_2_0-torch-1_6_0-py3_8:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-cpu-variant-tf-2_2_0-torch-1_6_0-py3_8 - push-retries: 5 - - docker-compose#v3.7.0: - push: test-cpu-variant-tf-2_2_0-torch-1_6_0-py3_8:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-cpu-variant-tf-2_2_0-torch-1_6_0-py3_8 - config: docker-compose.test.yml - - docker-compose#v3.7.0: - run: test-cpu-variant-tf-2_2_0-torch-1_6_0-py3_8 - config: docker-compose.test.yml - env: - - NEUROPOD_CACHE_ACCESS_KEY - - NEUROPOD_CACHE_ACCESS_SECRET - - BUILDKITE - - BUILDKITE_BRANCH - - BUILDKITE_BUILD_NUMBER - - BUILDKITE_BUILD_URL - - BUILDKITE_COMMIT - - BUILDKITE_JOB_ID - - BUILDKITE_PROJECT_SLUG - - BUILDKITE_PULL_REQUEST - - BUILDKITE_TAG - - CI - - CODECOV_TOKEN - - GH_STATUS_TOKEN - - GH_UPLOAD_TOKEN - - NEUROPOD_TEST_FRAMEWORKS - - WEB_DEPLOY_KEY - retry: - automatic: true - - - label: ":docker: CPU Tests (test-cpu-variant-tf-2_2_0-torch-1_7_0-py3_8)" - timeout_in_minutes: 60 - agents: - queue: public-gpu - env: - NEUROPOD_TEST_FRAMEWORKS: torchscript,python - command: build/ci/buildkite_build.sh - plugins: - - docker-compose#v3.7.0: - build: test-cpu-variant-tf-2_2_0-torch-1_7_0-py3_8 - config: docker-compose.test.yml - image-repository: 027047743804.dkr.ecr.us-east-2.amazonaws.com/uber - cache-from: test-cpu-variant-tf-2_2_0-torch-1_7_0-py3_8:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-cpu-variant-tf-2_2_0-torch-1_7_0-py3_8 - push-retries: 5 - - docker-compose#v3.7.0: - push: test-cpu-variant-tf-2_2_0-torch-1_7_0-py3_8:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-cpu-variant-tf-2_2_0-torch-1_7_0-py3_8 - config: docker-compose.test.yml - - docker-compose#v3.7.0: - run: test-cpu-variant-tf-2_2_0-torch-1_7_0-py3_8 - config: docker-compose.test.yml - env: - - NEUROPOD_CACHE_ACCESS_KEY - - NEUROPOD_CACHE_ACCESS_SECRET - - BUILDKITE - - BUILDKITE_BRANCH - - BUILDKITE_BUILD_NUMBER - - BUILDKITE_BUILD_URL - - BUILDKITE_COMMIT - - BUILDKITE_JOB_ID - - BUILDKITE_PROJECT_SLUG - - BUILDKITE_PULL_REQUEST - - BUILDKITE_TAG - - CI - - CODECOV_TOKEN - - GH_STATUS_TOKEN - - GH_UPLOAD_TOKEN - - NEUROPOD_TEST_FRAMEWORKS - - WEB_DEPLOY_KEY - retry: - automatic: true - - - label: ":docker: CPU Tests (test-cpu-variant-tf-2_2_0-torch-1_8_1-py3_8)" - timeout_in_minutes: 60 - agents: - queue: public-gpu - env: - NEUROPOD_TEST_FRAMEWORKS: torchscript,python - command: build/ci/buildkite_build.sh - plugins: - - docker-compose#v3.7.0: - build: test-cpu-variant-tf-2_2_0-torch-1_8_1-py3_8 - config: docker-compose.test.yml - image-repository: 027047743804.dkr.ecr.us-east-2.amazonaws.com/uber - cache-from: test-cpu-variant-tf-2_2_0-torch-1_8_1-py3_8:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-cpu-variant-tf-2_2_0-torch-1_8_1-py3_8 - push-retries: 5 - - docker-compose#v3.7.0: - push: test-cpu-variant-tf-2_2_0-torch-1_8_1-py3_8:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-cpu-variant-tf-2_2_0-torch-1_8_1-py3_8 - config: docker-compose.test.yml - - docker-compose#v3.7.0: - run: test-cpu-variant-tf-2_2_0-torch-1_8_1-py3_8 - config: docker-compose.test.yml - env: - - NEUROPOD_CACHE_ACCESS_KEY - - NEUROPOD_CACHE_ACCESS_SECRET - - BUILDKITE - - BUILDKITE_BRANCH - - BUILDKITE_BUILD_NUMBER - - BUILDKITE_BUILD_URL - - BUILDKITE_COMMIT - - BUILDKITE_JOB_ID - - BUILDKITE_PROJECT_SLUG - - BUILDKITE_PULL_REQUEST - - BUILDKITE_TAG - - CI - - CODECOV_TOKEN - - GH_STATUS_TOKEN - - GH_UPLOAD_TOKEN - - NEUROPOD_TEST_FRAMEWORKS - - WEB_DEPLOY_KEY - retry: - automatic: true - - - label: ":docker: CPU Tests (test-cpu-variant-tf-2_2_0-torch-1_9_0-py3_8)" - timeout_in_minutes: 60 - agents: - queue: public-gpu - env: - NEUROPOD_TEST_FRAMEWORKS: torchscript,python - command: build/ci/buildkite_build.sh - plugins: - - docker-compose#v3.7.0: - build: test-cpu-variant-tf-2_2_0-torch-1_9_0-py3_8 - config: docker-compose.test.yml - image-repository: 027047743804.dkr.ecr.us-east-2.amazonaws.com/uber - cache-from: test-cpu-variant-tf-2_2_0-torch-1_9_0-py3_8:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-cpu-variant-tf-2_2_0-torch-1_9_0-py3_8 - push-retries: 5 - - docker-compose#v3.7.0: - push: test-cpu-variant-tf-2_2_0-torch-1_9_0-py3_8:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-cpu-variant-tf-2_2_0-torch-1_9_0-py3_8 - config: docker-compose.test.yml - - docker-compose#v3.7.0: - run: test-cpu-variant-tf-2_2_0-torch-1_9_0-py3_8 - config: docker-compose.test.yml - env: - - NEUROPOD_CACHE_ACCESS_KEY - - NEUROPOD_CACHE_ACCESS_SECRET - - BUILDKITE - - BUILDKITE_BRANCH - - BUILDKITE_BUILD_NUMBER - - BUILDKITE_BUILD_URL - - BUILDKITE_COMMIT - - BUILDKITE_JOB_ID - - BUILDKITE_PROJECT_SLUG - - BUILDKITE_PULL_REQUEST - - BUILDKITE_TAG - - CI - - CODECOV_TOKEN - - GH_STATUS_TOKEN - - GH_UPLOAD_TOKEN - - NEUROPOD_TEST_FRAMEWORKS - - WEB_DEPLOY_KEY - retry: - automatic: true - - - label: ":docker: CPU Tests (test-cpu-variant-tf-2_2_0-torch-1_10_2-py3_8)" - timeout_in_minutes: 60 - agents: - queue: public-gpu - env: - NEUROPOD_TEST_FRAMEWORKS: torchscript,python - command: build/ci/buildkite_build.sh - plugins: - - docker-compose#v3.7.0: - build: test-cpu-variant-tf-2_2_0-torch-1_10_2-py3_8 + build: test-cpu-variant-tf-2_5_0-torch-1_7_1-py3_9 config: docker-compose.test.yml image-repository: 027047743804.dkr.ecr.us-east-2.amazonaws.com/uber - cache-from: test-cpu-variant-tf-2_2_0-torch-1_10_2-py3_8:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-cpu-variant-tf-2_2_0-torch-1_10_2-py3_8 + cache-from: test-cpu-variant-tf-2_5_0-torch-1_7_1-py3_9:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-cpu-variant-tf-2_5_0-torch-1_7_1-py3_9 push-retries: 5 - docker-compose#v3.7.0: - push: test-cpu-variant-tf-2_2_0-torch-1_10_2-py3_8:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-cpu-variant-tf-2_2_0-torch-1_10_2-py3_8 + push: test-cpu-variant-tf-2_5_0-torch-1_7_1-py3_9:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-cpu-variant-tf-2_5_0-torch-1_7_1-py3_9 config: docker-compose.test.yml - docker-compose#v3.7.0: - run: test-cpu-variant-tf-2_2_0-torch-1_10_2-py3_8 + run: test-cpu-variant-tf-2_5_0-torch-1_7_1-py3_9 config: docker-compose.test.yml env: - NEUROPOD_CACHE_ACCESS_KEY @@ -372,7 +85,7 @@ steps: retry: automatic: true - - label: ":docker: CPU Tests (test-cpu-variant-tf-2_5_0-torch-1_7_0-py3_8)" + - label: ":docker: CPU Tests (test-cpu-variant-tf-2_6_2-torch-1_7_1-py3_9)" timeout_in_minutes: 60 agents: queue: public-gpu @@ -381,385 +94,16 @@ steps: command: build/ci/buildkite_build.sh plugins: - docker-compose#v3.7.0: - build: test-cpu-variant-tf-2_5_0-torch-1_7_0-py3_8 - config: docker-compose.test.yml - image-repository: 027047743804.dkr.ecr.us-east-2.amazonaws.com/uber - cache-from: test-cpu-variant-tf-2_5_0-torch-1_7_0-py3_8:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-cpu-variant-tf-2_5_0-torch-1_7_0-py3_8 - push-retries: 5 - - docker-compose#v3.7.0: - push: test-cpu-variant-tf-2_5_0-torch-1_7_0-py3_8:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-cpu-variant-tf-2_5_0-torch-1_7_0-py3_8 - config: docker-compose.test.yml - - docker-compose#v3.7.0: - run: test-cpu-variant-tf-2_5_0-torch-1_7_0-py3_8 - config: docker-compose.test.yml - env: - - NEUROPOD_CACHE_ACCESS_KEY - - NEUROPOD_CACHE_ACCESS_SECRET - - BUILDKITE - - BUILDKITE_BRANCH - - BUILDKITE_BUILD_NUMBER - - BUILDKITE_BUILD_URL - - BUILDKITE_COMMIT - - BUILDKITE_JOB_ID - - BUILDKITE_PROJECT_SLUG - - BUILDKITE_PULL_REQUEST - - BUILDKITE_TAG - - CI - - CODECOV_TOKEN - - GH_STATUS_TOKEN - - GH_UPLOAD_TOKEN - - NEUROPOD_TEST_FRAMEWORKS - - WEB_DEPLOY_KEY - retry: - automatic: true - - - label: ":docker: CPU Tests (test-cpu-variant-tf-2_6_2-torch-1_7_0-py3_8)" - timeout_in_minutes: 60 - agents: - queue: public-gpu - env: - NEUROPOD_TEST_FRAMEWORKS: tensorflow - command: build/ci/buildkite_build.sh - plugins: - - docker-compose#v3.7.0: - build: test-cpu-variant-tf-2_6_2-torch-1_7_0-py3_8 - config: docker-compose.test.yml - image-repository: 027047743804.dkr.ecr.us-east-2.amazonaws.com/uber - cache-from: test-cpu-variant-tf-2_6_2-torch-1_7_0-py3_8:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-cpu-variant-tf-2_6_2-torch-1_7_0-py3_8 - push-retries: 5 - - docker-compose#v3.7.0: - push: test-cpu-variant-tf-2_6_2-torch-1_7_0-py3_8:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-cpu-variant-tf-2_6_2-torch-1_7_0-py3_8 - config: docker-compose.test.yml - - docker-compose#v3.7.0: - run: test-cpu-variant-tf-2_6_2-torch-1_7_0-py3_8 - config: docker-compose.test.yml - env: - - NEUROPOD_CACHE_ACCESS_KEY - - NEUROPOD_CACHE_ACCESS_SECRET - - BUILDKITE - - BUILDKITE_BRANCH - - BUILDKITE_BUILD_NUMBER - - BUILDKITE_BUILD_URL - - BUILDKITE_COMMIT - - BUILDKITE_JOB_ID - - BUILDKITE_PROJECT_SLUG - - BUILDKITE_PULL_REQUEST - - BUILDKITE_TAG - - CI - - CODECOV_TOKEN - - GH_STATUS_TOKEN - - GH_UPLOAD_TOKEN - - NEUROPOD_TEST_FRAMEWORKS - - WEB_DEPLOY_KEY - retry: - automatic: true - - - label: ":docker: GPU Tests (test-gpu-variant-tf-1_14_0-torch-1_3_0-py3_6)" - timeout_in_minutes: 60 - agents: - queue: public-gpu - env: - NEUROPOD_TEST_FRAMEWORKS: tensorflow,torchscript,python - command: build/ci/buildkite_build_gpu.sh - plugins: - - docker-compose#v3.7.0: - build: test-gpu-variant-tf-1_14_0-torch-1_3_0-py3_6 - config: docker-compose.test.yml - image-repository: 027047743804.dkr.ecr.us-east-2.amazonaws.com/uber - cache-from: test-gpu-variant-tf-1_14_0-torch-1_3_0-py3_6:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-gpu-variant-tf-1_14_0-torch-1_3_0-py3_6 - push-retries: 5 - - docker-compose#v3.7.0: - push: test-gpu-variant-tf-1_14_0-torch-1_3_0-py3_6:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-gpu-variant-tf-1_14_0-torch-1_3_0-py3_6 - config: docker-compose.test.yml - - docker-compose#v3.7.0: - run: test-gpu-variant-tf-1_14_0-torch-1_3_0-py3_6 - config: docker-compose.test.yml - env: - - NEUROPOD_CACHE_ACCESS_KEY - - NEUROPOD_CACHE_ACCESS_SECRET - - BUILDKITE - - BUILDKITE_BRANCH - - BUILDKITE_BUILD_NUMBER - - BUILDKITE_BUILD_URL - - BUILDKITE_COMMIT - - BUILDKITE_JOB_ID - - BUILDKITE_PROJECT_SLUG - - BUILDKITE_PULL_REQUEST - - BUILDKITE_TAG - - CI - - CODECOV_TOKEN - - GH_STATUS_TOKEN - - GH_UPLOAD_TOKEN - - NEUROPOD_TEST_FRAMEWORKS - - WEB_DEPLOY_KEY - retry: - automatic: true - - - label: ":docker: GPU Tests (test-gpu-variant-tf-1_15_0-torch-1_4_0-py3_7)" - timeout_in_minutes: 60 - agents: - queue: public-gpu - env: - NEUROPOD_TEST_FRAMEWORKS: tensorflow,torchscript,python - command: build/ci/buildkite_build_gpu.sh - plugins: - - docker-compose#v3.7.0: - build: test-gpu-variant-tf-1_15_0-torch-1_4_0-py3_7 - config: docker-compose.test.yml - image-repository: 027047743804.dkr.ecr.us-east-2.amazonaws.com/uber - cache-from: test-gpu-variant-tf-1_15_0-torch-1_4_0-py3_7:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-gpu-variant-tf-1_15_0-torch-1_4_0-py3_7 - push-retries: 5 - - docker-compose#v3.7.0: - push: test-gpu-variant-tf-1_15_0-torch-1_4_0-py3_7:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-gpu-variant-tf-1_15_0-torch-1_4_0-py3_7 - config: docker-compose.test.yml - - docker-compose#v3.7.0: - run: test-gpu-variant-tf-1_15_0-torch-1_4_0-py3_7 - config: docker-compose.test.yml - env: - - NEUROPOD_CACHE_ACCESS_KEY - - NEUROPOD_CACHE_ACCESS_SECRET - - BUILDKITE - - BUILDKITE_BRANCH - - BUILDKITE_BUILD_NUMBER - - BUILDKITE_BUILD_URL - - BUILDKITE_COMMIT - - BUILDKITE_JOB_ID - - BUILDKITE_PROJECT_SLUG - - BUILDKITE_PULL_REQUEST - - BUILDKITE_TAG - - CI - - CODECOV_TOKEN - - GH_STATUS_TOKEN - - GH_UPLOAD_TOKEN - - NEUROPOD_TEST_FRAMEWORKS - - WEB_DEPLOY_KEY - retry: - automatic: true - - - label: ":docker: GPU Tests (test-gpu-variant-tf-2_2_0-torch-1_5_0-py3_8)" - timeout_in_minutes: 60 - agents: - queue: public-gpu - env: - NEUROPOD_TEST_FRAMEWORKS: tensorflow,torchscript,python - command: build/ci/buildkite_build_gpu.sh - plugins: - - docker-compose#v3.7.0: - build: test-gpu-variant-tf-2_2_0-torch-1_5_0-py3_8 - config: docker-compose.test.yml - image-repository: 027047743804.dkr.ecr.us-east-2.amazonaws.com/uber - cache-from: test-gpu-variant-tf-2_2_0-torch-1_5_0-py3_8:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-gpu-variant-tf-2_2_0-torch-1_5_0-py3_8 - push-retries: 5 - - docker-compose#v3.7.0: - push: test-gpu-variant-tf-2_2_0-torch-1_5_0-py3_8:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-gpu-variant-tf-2_2_0-torch-1_5_0-py3_8 - config: docker-compose.test.yml - - docker-compose#v3.7.0: - run: test-gpu-variant-tf-2_2_0-torch-1_5_0-py3_8 - config: docker-compose.test.yml - env: - - NEUROPOD_CACHE_ACCESS_KEY - - NEUROPOD_CACHE_ACCESS_SECRET - - BUILDKITE - - BUILDKITE_BRANCH - - BUILDKITE_BUILD_NUMBER - - BUILDKITE_BUILD_URL - - BUILDKITE_COMMIT - - BUILDKITE_JOB_ID - - BUILDKITE_PROJECT_SLUG - - BUILDKITE_PULL_REQUEST - - BUILDKITE_TAG - - CI - - CODECOV_TOKEN - - GH_STATUS_TOKEN - - GH_UPLOAD_TOKEN - - NEUROPOD_TEST_FRAMEWORKS - - WEB_DEPLOY_KEY - retry: - automatic: true - - - label: ":docker: GPU Tests (test-gpu-variant-tf-2_2_0-torch-1_6_0-py3_8)" - timeout_in_minutes: 60 - agents: - queue: public-gpu - env: - NEUROPOD_TEST_FRAMEWORKS: torchscript,python - command: build/ci/buildkite_build_gpu.sh - plugins: - - docker-compose#v3.7.0: - build: test-gpu-variant-tf-2_2_0-torch-1_6_0-py3_8 - config: docker-compose.test.yml - image-repository: 027047743804.dkr.ecr.us-east-2.amazonaws.com/uber - cache-from: test-gpu-variant-tf-2_2_0-torch-1_6_0-py3_8:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-gpu-variant-tf-2_2_0-torch-1_6_0-py3_8 - push-retries: 5 - - docker-compose#v3.7.0: - push: test-gpu-variant-tf-2_2_0-torch-1_6_0-py3_8:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-gpu-variant-tf-2_2_0-torch-1_6_0-py3_8 - config: docker-compose.test.yml - - docker-compose#v3.7.0: - run: test-gpu-variant-tf-2_2_0-torch-1_6_0-py3_8 - config: docker-compose.test.yml - env: - - NEUROPOD_CACHE_ACCESS_KEY - - NEUROPOD_CACHE_ACCESS_SECRET - - BUILDKITE - - BUILDKITE_BRANCH - - BUILDKITE_BUILD_NUMBER - - BUILDKITE_BUILD_URL - - BUILDKITE_COMMIT - - BUILDKITE_JOB_ID - - BUILDKITE_PROJECT_SLUG - - BUILDKITE_PULL_REQUEST - - BUILDKITE_TAG - - CI - - CODECOV_TOKEN - - GH_STATUS_TOKEN - - GH_UPLOAD_TOKEN - - NEUROPOD_TEST_FRAMEWORKS - - WEB_DEPLOY_KEY - retry: - automatic: true - - - label: ":docker: GPU Tests (test-gpu-variant-tf-2_2_0-torch-1_7_0-py3_8)" - timeout_in_minutes: 60 - agents: - queue: public-gpu - env: - NEUROPOD_TEST_FRAMEWORKS: torchscript,python - command: build/ci/buildkite_build_gpu.sh - plugins: - - docker-compose#v3.7.0: - build: test-gpu-variant-tf-2_2_0-torch-1_7_0-py3_8 - config: docker-compose.test.yml - image-repository: 027047743804.dkr.ecr.us-east-2.amazonaws.com/uber - cache-from: test-gpu-variant-tf-2_2_0-torch-1_7_0-py3_8:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-gpu-variant-tf-2_2_0-torch-1_7_0-py3_8 - push-retries: 5 - - docker-compose#v3.7.0: - push: test-gpu-variant-tf-2_2_0-torch-1_7_0-py3_8:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-gpu-variant-tf-2_2_0-torch-1_7_0-py3_8 - config: docker-compose.test.yml - - docker-compose#v3.7.0: - run: test-gpu-variant-tf-2_2_0-torch-1_7_0-py3_8 - config: docker-compose.test.yml - env: - - NEUROPOD_CACHE_ACCESS_KEY - - NEUROPOD_CACHE_ACCESS_SECRET - - BUILDKITE - - BUILDKITE_BRANCH - - BUILDKITE_BUILD_NUMBER - - BUILDKITE_BUILD_URL - - BUILDKITE_COMMIT - - BUILDKITE_JOB_ID - - BUILDKITE_PROJECT_SLUG - - BUILDKITE_PULL_REQUEST - - BUILDKITE_TAG - - CI - - CODECOV_TOKEN - - GH_STATUS_TOKEN - - GH_UPLOAD_TOKEN - - NEUROPOD_TEST_FRAMEWORKS - - WEB_DEPLOY_KEY - retry: - automatic: true - - - label: ":docker: GPU Tests (test-gpu-variant-tf-2_2_0-torch-1_8_1-py3_8)" - timeout_in_minutes: 60 - agents: - queue: public-gpu - env: - NEUROPOD_TEST_FRAMEWORKS: torchscript,python - command: build/ci/buildkite_build_gpu.sh - plugins: - - docker-compose#v3.7.0: - build: test-gpu-variant-tf-2_2_0-torch-1_8_1-py3_8 - config: docker-compose.test.yml - image-repository: 027047743804.dkr.ecr.us-east-2.amazonaws.com/uber - cache-from: test-gpu-variant-tf-2_2_0-torch-1_8_1-py3_8:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-gpu-variant-tf-2_2_0-torch-1_8_1-py3_8 - push-retries: 5 - - docker-compose#v3.7.0: - push: test-gpu-variant-tf-2_2_0-torch-1_8_1-py3_8:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-gpu-variant-tf-2_2_0-torch-1_8_1-py3_8 - config: docker-compose.test.yml - - docker-compose#v3.7.0: - run: test-gpu-variant-tf-2_2_0-torch-1_8_1-py3_8 - config: docker-compose.test.yml - env: - - NEUROPOD_CACHE_ACCESS_KEY - - NEUROPOD_CACHE_ACCESS_SECRET - - BUILDKITE - - BUILDKITE_BRANCH - - BUILDKITE_BUILD_NUMBER - - BUILDKITE_BUILD_URL - - BUILDKITE_COMMIT - - BUILDKITE_JOB_ID - - BUILDKITE_PROJECT_SLUG - - BUILDKITE_PULL_REQUEST - - BUILDKITE_TAG - - CI - - CODECOV_TOKEN - - GH_STATUS_TOKEN - - GH_UPLOAD_TOKEN - - NEUROPOD_TEST_FRAMEWORKS - - WEB_DEPLOY_KEY - retry: - automatic: true - - - label: ":docker: GPU Tests (test-gpu-variant-tf-2_2_0-torch-1_9_0-py3_8)" - timeout_in_minutes: 60 - agents: - queue: public-gpu - env: - NEUROPOD_TEST_FRAMEWORKS: torchscript,python - command: build/ci/buildkite_build_gpu.sh - plugins: - - docker-compose#v3.7.0: - build: test-gpu-variant-tf-2_2_0-torch-1_9_0-py3_8 - config: docker-compose.test.yml - image-repository: 027047743804.dkr.ecr.us-east-2.amazonaws.com/uber - cache-from: test-gpu-variant-tf-2_2_0-torch-1_9_0-py3_8:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-gpu-variant-tf-2_2_0-torch-1_9_0-py3_8 - push-retries: 5 - - docker-compose#v3.7.0: - push: test-gpu-variant-tf-2_2_0-torch-1_9_0-py3_8:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-gpu-variant-tf-2_2_0-torch-1_9_0-py3_8 - config: docker-compose.test.yml - - docker-compose#v3.7.0: - run: test-gpu-variant-tf-2_2_0-torch-1_9_0-py3_8 - config: docker-compose.test.yml - env: - - NEUROPOD_CACHE_ACCESS_KEY - - NEUROPOD_CACHE_ACCESS_SECRET - - BUILDKITE - - BUILDKITE_BRANCH - - BUILDKITE_BUILD_NUMBER - - BUILDKITE_BUILD_URL - - BUILDKITE_COMMIT - - BUILDKITE_JOB_ID - - BUILDKITE_PROJECT_SLUG - - BUILDKITE_PULL_REQUEST - - BUILDKITE_TAG - - CI - - CODECOV_TOKEN - - GH_STATUS_TOKEN - - GH_UPLOAD_TOKEN - - NEUROPOD_TEST_FRAMEWORKS - - WEB_DEPLOY_KEY - retry: - automatic: true - - - label: ":docker: GPU Tests (test-gpu-variant-tf-2_2_0-torch-1_10_2-py3_8)" - timeout_in_minutes: 60 - agents: - queue: public-gpu - env: - NEUROPOD_TEST_FRAMEWORKS: torchscript,python - command: build/ci/buildkite_build_gpu.sh - plugins: - - docker-compose#v3.7.0: - build: test-gpu-variant-tf-2_2_0-torch-1_10_2-py3_8 + build: test-cpu-variant-tf-2_6_2-torch-1_7_1-py3_9 config: docker-compose.test.yml image-repository: 027047743804.dkr.ecr.us-east-2.amazonaws.com/uber - cache-from: test-gpu-variant-tf-2_2_0-torch-1_10_2-py3_8:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-gpu-variant-tf-2_2_0-torch-1_10_2-py3_8 + cache-from: test-cpu-variant-tf-2_6_2-torch-1_7_1-py3_9:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-cpu-variant-tf-2_6_2-torch-1_7_1-py3_9 push-retries: 5 - docker-compose#v3.7.0: - push: test-gpu-variant-tf-2_2_0-torch-1_10_2-py3_8:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-gpu-variant-tf-2_2_0-torch-1_10_2-py3_8 + push: test-cpu-variant-tf-2_6_2-torch-1_7_1-py3_9:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-cpu-variant-tf-2_6_2-torch-1_7_1-py3_9 config: docker-compose.test.yml - docker-compose#v3.7.0: - run: test-gpu-variant-tf-2_2_0-torch-1_10_2-py3_8 + run: test-cpu-variant-tf-2_6_2-torch-1_7_1-py3_9 config: docker-compose.test.yml env: - NEUROPOD_CACHE_ACCESS_KEY @@ -782,7 +126,7 @@ steps: retry: automatic: true - - label: ":docker: GPU Tests (test-gpu-variant-tf-2_5_0-torch-1_7_0-py3_8)" + - label: ":docker: GPU Tests (test-gpu-variant-tf-2_5_0-torch-1_7_1-py3_9)" timeout_in_minutes: 60 agents: queue: public-gpu @@ -791,16 +135,16 @@ steps: command: build/ci/buildkite_build_gpu.sh plugins: - docker-compose#v3.7.0: - build: test-gpu-variant-tf-2_5_0-torch-1_7_0-py3_8 + build: test-gpu-variant-tf-2_5_0-torch-1_7_1-py3_9 config: docker-compose.test.yml image-repository: 027047743804.dkr.ecr.us-east-2.amazonaws.com/uber - cache-from: test-gpu-variant-tf-2_5_0-torch-1_7_0-py3_8:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-gpu-variant-tf-2_5_0-torch-1_7_0-py3_8 + cache-from: test-gpu-variant-tf-2_5_0-torch-1_7_1-py3_9:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-gpu-variant-tf-2_5_0-torch-1_7_1-py3_9 push-retries: 5 - docker-compose#v3.7.0: - push: test-gpu-variant-tf-2_5_0-torch-1_7_0-py3_8:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-gpu-variant-tf-2_5_0-torch-1_7_0-py3_8 + push: test-gpu-variant-tf-2_5_0-torch-1_7_1-py3_9:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-gpu-variant-tf-2_5_0-torch-1_7_1-py3_9 config: docker-compose.test.yml - docker-compose#v3.7.0: - run: test-gpu-variant-tf-2_5_0-torch-1_7_0-py3_8 + run: test-gpu-variant-tf-2_5_0-torch-1_7_1-py3_9 config: docker-compose.test.yml env: - NEUROPOD_CACHE_ACCESS_KEY @@ -823,7 +167,7 @@ steps: retry: automatic: true - - label: ":docker: GPU Tests (test-gpu-variant-tf-2_6_2-torch-1_7_0-py3_8)" + - label: ":docker: GPU Tests (test-gpu-variant-tf-2_6_2-torch-1_7_1-py3_9)" timeout_in_minutes: 60 agents: queue: public-gpu @@ -832,16 +176,16 @@ steps: command: build/ci/buildkite_build_gpu.sh plugins: - docker-compose#v3.7.0: - build: test-gpu-variant-tf-2_6_2-torch-1_7_0-py3_8 + build: test-gpu-variant-tf-2_6_2-torch-1_7_1-py3_9 config: docker-compose.test.yml image-repository: 027047743804.dkr.ecr.us-east-2.amazonaws.com/uber - cache-from: test-gpu-variant-tf-2_6_2-torch-1_7_0-py3_8:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-gpu-variant-tf-2_6_2-torch-1_7_0-py3_8 + cache-from: test-gpu-variant-tf-2_6_2-torch-1_7_1-py3_9:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-gpu-variant-tf-2_6_2-torch-1_7_1-py3_9 push-retries: 5 - docker-compose#v3.7.0: - push: test-gpu-variant-tf-2_6_2-torch-1_7_0-py3_8:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-gpu-variant-tf-2_6_2-torch-1_7_0-py3_8 + push: test-gpu-variant-tf-2_6_2-torch-1_7_1-py3_9:027047743804.dkr.ecr.us-east-2.amazonaws.com/uber:test-gpu-variant-tf-2_6_2-torch-1_7_1-py3_9 config: docker-compose.test.yml - docker-compose#v3.7.0: - run: test-gpu-variant-tf-2_6_2-torch-1_7_0-py3_8 + run: test-gpu-variant-tf-2_6_2-torch-1_7_1-py3_9 config: docker-compose.test.yml env: - NEUROPOD_CACHE_ACCESS_KEY diff --git a/.github/workflows/mac_ci.yml b/.github/workflows/mac_ci.yml index 43f0ec21..503ac0e5 100644 --- a/.github/workflows/mac_ci.yml +++ b/.github/workflows/mac_ci.yml @@ -38,54 +38,14 @@ jobs: strategy: matrix: include: - - tf: 1.14.0 - torch: 1.3.0 - python: 3.6 - test_frameworks: tensorflow,torchscript,python - - - tf: 1.15.0 - torch: 1.4.0 - python: 3.7 - test_frameworks: tensorflow,torchscript,python - - - tf: 2.2.0 - torch: 1.5.0 - python: 3.8 - test_frameworks: tensorflow,torchscript,python - - - tf: 2.2.0 - torch: 1.6.0 - python: 3.8 - test_frameworks: torchscript,python - - - tf: 2.2.0 - torch: 1.7.0 - python: 3.8 - test_frameworks: torchscript,python - - - tf: 2.2.0 - torch: 1.8.1 - python: 3.8 - test_frameworks: torchscript,python - - - tf: 2.2.0 - torch: 1.9.0 - python: 3.8 - test_frameworks: torchscript,python - - - tf: 2.2.0 - torch: 1.10.2 - python: 3.8 - test_frameworks: torchscript,python - - tf: 2.5.0 - torch: 1.7.0 - python: 3.8 + torch: 1.7.1 + python: 3.9 test_frameworks: tensorflow - tf: 2.6.2 - torch: 1.7.0 - python: 3.8 + torch: 1.7.1 + python: 3.9 test_frameworks: tensorflow diff --git a/build/ci/gh_actions_build.sh b/build/ci/gh_actions_build.sh index e607f9db..a0508de2 100755 --- a/build/ci/gh_actions_build.sh +++ b/build/ci/gh_actions_build.sh @@ -33,6 +33,9 @@ if [[ $(uname -s) == 'Darwin' ]]; then elif [[ "${NEUROPOD_PYTHON_VERSION}" == "3.8" ]]; then wget https://www.python.org/ftp/python/3.8.2/python-3.8.2-macosx10.9.pkg &> /dev/null sudo installer -pkg python-3.8.2-macosx10.9.pkg -target / + elif [[ "${NEUROPOD_PYTHON_VERSION}" == "3.9" ]]; then + wget https://www.python.org/ftp/python/3.9.13/python-3.9.13-macosx10.9.pkg &> /dev/null + sudo installer -pkg python-3.9.13-macosx10.9.pkg -target / fi fi diff --git a/build/ci_matrix.py b/build/ci_matrix.py index 4eb7c99d..13eeddc8 100644 --- a/build/ci_matrix.py +++ b/build/ci_matrix.py @@ -104,20 +104,9 @@ # Versions of frameworks to test with FRAMEWORK_VERSIONS = [ - {"cuda": "10.0", "tensorflow": "1.14.0", "torch": "1.3.0", "python": "3.6"}, - {"cuda": "10.0", "tensorflow": "1.15.0", "torch": "1.4.0", "python": "3.7"}, - {"cuda": "10.1", "tensorflow": "2.2.0", "torch": "1.5.0", "python": "3.8"}, - - # No need to rerun tensorflow tests for 2.2.0 on py3.8 - {"cuda": "10.1", "tensorflow": "2.2.0", "torch": "1.6.0", "python": "3.8", "test_frameworks": "torchscript,python"}, - {"cuda": "10.1", "tensorflow": "2.2.0", "torch": "1.7.0", "python": "3.8", "test_frameworks": "torchscript,python"}, - {"cuda": "10.2", "tensorflow": "2.2.0", "torch": "1.8.1", "python": "3.8", "test_frameworks": "torchscript,python"}, - {"cuda": "10.2", "tensorflow": "2.2.0", "torch": "1.9.0", "python": "3.8", "test_frameworks": "torchscript,python"}, - {"cuda": "10.2", "tensorflow": "2.2.0", "torch": "1.10.2", "python": "3.8", "test_frameworks": "torchscript,python"}, - - # Only testing TF - {"cuda": "11.2.1", "cudnn": "8", "tensorflow": "2.5.0", "torch": "1.7.0", "python": "3.8", "test_frameworks": "tensorflow"}, - {"cuda": "11.2.1", "cudnn": "8", "tensorflow": "2.6.2", "torch": "1.7.0", "python": "3.8", "test_frameworks": "tensorflow"}, + # Only testing TF for Python 3.9 + {"cuda": "11.2.1", "cudnn": "8", "tensorflow": "2.5.0", "torch": "1.7.1", "python": "3.9", "test_frameworks": "tensorflow"}, + {"cuda": "11.2.1", "cudnn": "8", "tensorflow": "2.6.2", "torch": "1.7.1", "python": "3.9", "test_frameworks": "tensorflow"}, ] gh_actions_matrix = [] diff --git a/build/install_python_deps.sh b/build/install_python_deps.sh index 21abc0ba..d3dadd95 100755 --- a/build/install_python_deps.sh +++ b/build/install_python_deps.sh @@ -9,7 +9,7 @@ wget https://bootstrap.pypa.io/pip/3.6/get-pip.py -O /tmp/get-pip.py ${NEUROPOD_PYTHON_BINARY} /tmp/get-pip.py # Setup a virtualenv -${NEUROPOD_PYTHON_BINARY} -m pip install virtualenv==16.7.9 +${NEUROPOD_PYTHON_BINARY} -m pip install virtualenv ${NEUROPOD_PYTHON_BINARY} -m virtualenv .neuropod_venv source .neuropod_venv/bin/activate diff --git a/docker-compose.test.yml b/docker-compose.test.yml index 51fe6fc1..25b9466e 100644 --- a/docker-compose.test.yml +++ b/docker-compose.test.yml @@ -19,176 +19,40 @@ services: NEUROPOD_IS_GPU: "true" runtime: nvidia - test-cpu-variant-tf-1_14_0-torch-1_3_0-py3_6: - extends: test-base - build: - args: - NEUROPOD_TENSORFLOW_VERSION: 1.14.0 - NEUROPOD_TORCH_VERSION: 1.3.0 - NEUROPOD_PYTHON_VERSION: 3.6 - - test-cpu-variant-tf-1_15_0-torch-1_4_0-py3_7: - extends: test-base - build: - args: - NEUROPOD_TENSORFLOW_VERSION: 1.15.0 - NEUROPOD_TORCH_VERSION: 1.4.0 - NEUROPOD_PYTHON_VERSION: 3.7 - - test-cpu-variant-tf-2_2_0-torch-1_5_0-py3_8: - extends: test-base - build: - args: - NEUROPOD_TENSORFLOW_VERSION: 2.2.0 - NEUROPOD_TORCH_VERSION: 1.5.0 - NEUROPOD_PYTHON_VERSION: 3.8 - - test-cpu-variant-tf-2_2_0-torch-1_6_0-py3_8: - extends: test-base - build: - args: - NEUROPOD_TENSORFLOW_VERSION: 2.2.0 - NEUROPOD_TORCH_VERSION: 1.6.0 - NEUROPOD_PYTHON_VERSION: 3.8 - - test-cpu-variant-tf-2_2_0-torch-1_7_0-py3_8: - extends: test-base - build: - args: - NEUROPOD_TENSORFLOW_VERSION: 2.2.0 - NEUROPOD_TORCH_VERSION: 1.7.0 - NEUROPOD_PYTHON_VERSION: 3.8 - - test-cpu-variant-tf-2_2_0-torch-1_8_1-py3_8: - extends: test-base - build: - args: - NEUROPOD_TENSORFLOW_VERSION: 2.2.0 - NEUROPOD_TORCH_VERSION: 1.8.1 - NEUROPOD_PYTHON_VERSION: 3.8 - - test-cpu-variant-tf-2_2_0-torch-1_9_0-py3_8: - extends: test-base - build: - args: - NEUROPOD_TENSORFLOW_VERSION: 2.2.0 - NEUROPOD_TORCH_VERSION: 1.9.0 - NEUROPOD_PYTHON_VERSION: 3.8 - - test-cpu-variant-tf-2_2_0-torch-1_10_2-py3_8: - extends: test-base - build: - args: - NEUROPOD_TENSORFLOW_VERSION: 2.2.0 - NEUROPOD_TORCH_VERSION: 1.10.2 - NEUROPOD_PYTHON_VERSION: 3.8 - - test-cpu-variant-tf-2_5_0-torch-1_7_0-py3_8: + test-cpu-variant-tf-2_5_0-torch-1_7_1-py3_9: extends: test-base build: args: NEUROPOD_TENSORFLOW_VERSION: 2.5.0 - NEUROPOD_TORCH_VERSION: 1.7.0 - NEUROPOD_PYTHON_VERSION: 3.8 + NEUROPOD_TORCH_VERSION: 1.7.1 + NEUROPOD_PYTHON_VERSION: 3.9 - test-cpu-variant-tf-2_6_2-torch-1_7_0-py3_8: + test-cpu-variant-tf-2_6_2-torch-1_7_1-py3_9: extends: test-base build: args: NEUROPOD_TENSORFLOW_VERSION: 2.6.2 - NEUROPOD_TORCH_VERSION: 1.7.0 - NEUROPOD_PYTHON_VERSION: 3.8 - - test-gpu-variant-tf-1_14_0-torch-1_3_0-py3_6: - extends: test-gpu - build: - args: - NEUROPOD_CUDA_VERSION: 10.0 - NEUROPOD_TENSORFLOW_VERSION: 1.14.0 - NEUROPOD_TORCH_VERSION: 1.3.0 - NEUROPOD_PYTHON_VERSION: 3.6 - - test-gpu-variant-tf-1_15_0-torch-1_4_0-py3_7: - extends: test-gpu - build: - args: - NEUROPOD_CUDA_VERSION: 10.0 - NEUROPOD_TENSORFLOW_VERSION: 1.15.0 - NEUROPOD_TORCH_VERSION: 1.4.0 - NEUROPOD_PYTHON_VERSION: 3.7 - - test-gpu-variant-tf-2_2_0-torch-1_5_0-py3_8: - extends: test-gpu - build: - args: - NEUROPOD_CUDA_VERSION: 10.1 - NEUROPOD_TENSORFLOW_VERSION: 2.2.0 - NEUROPOD_TORCH_VERSION: 1.5.0 - NEUROPOD_PYTHON_VERSION: 3.8 - - test-gpu-variant-tf-2_2_0-torch-1_6_0-py3_8: - extends: test-gpu - build: - args: - NEUROPOD_CUDA_VERSION: 10.1 - NEUROPOD_TENSORFLOW_VERSION: 2.2.0 - NEUROPOD_TORCH_VERSION: 1.6.0 - NEUROPOD_PYTHON_VERSION: 3.8 - - test-gpu-variant-tf-2_2_0-torch-1_7_0-py3_8: - extends: test-gpu - build: - args: - NEUROPOD_CUDA_VERSION: 10.1 - NEUROPOD_TENSORFLOW_VERSION: 2.2.0 - NEUROPOD_TORCH_VERSION: 1.7.0 - NEUROPOD_PYTHON_VERSION: 3.8 - - test-gpu-variant-tf-2_2_0-torch-1_8_1-py3_8: - extends: test-gpu - build: - args: - NEUROPOD_CUDA_VERSION: 10.2 - NEUROPOD_TENSORFLOW_VERSION: 2.2.0 - NEUROPOD_TORCH_VERSION: 1.8.1 - NEUROPOD_PYTHON_VERSION: 3.8 - - test-gpu-variant-tf-2_2_0-torch-1_9_0-py3_8: - extends: test-gpu - build: - args: - NEUROPOD_CUDA_VERSION: 10.2 - NEUROPOD_TENSORFLOW_VERSION: 2.2.0 - NEUROPOD_TORCH_VERSION: 1.9.0 - NEUROPOD_PYTHON_VERSION: 3.8 - - test-gpu-variant-tf-2_2_0-torch-1_10_2-py3_8: - extends: test-gpu - build: - args: - NEUROPOD_CUDA_VERSION: 10.2 - NEUROPOD_TENSORFLOW_VERSION: 2.2.0 - NEUROPOD_TORCH_VERSION: 1.10.2 - NEUROPOD_PYTHON_VERSION: 3.8 + NEUROPOD_TORCH_VERSION: 1.7.1 + NEUROPOD_PYTHON_VERSION: 3.9 - test-gpu-variant-tf-2_5_0-torch-1_7_0-py3_8: + test-gpu-variant-tf-2_5_0-torch-1_7_1-py3_9: extends: test-gpu build: args: NEUROPOD_CUDA_VERSION: 11.2.1 NEUROPOD_CUDNN_VERSION: 8 NEUROPOD_TENSORFLOW_VERSION: 2.5.0 - NEUROPOD_TORCH_VERSION: 1.7.0 - NEUROPOD_PYTHON_VERSION: 3.8 + NEUROPOD_TORCH_VERSION: 1.7.1 + NEUROPOD_PYTHON_VERSION: 3.9 - test-gpu-variant-tf-2_6_2-torch-1_7_0-py3_8: + test-gpu-variant-tf-2_6_2-torch-1_7_1-py3_9: extends: test-gpu build: args: NEUROPOD_CUDA_VERSION: 11.2.1 NEUROPOD_CUDNN_VERSION: 8 NEUROPOD_TENSORFLOW_VERSION: 2.6.2 - NEUROPOD_TORCH_VERSION: 1.7.0 - NEUROPOD_PYTHON_VERSION: 3.8 + NEUROPOD_TORCH_VERSION: 1.7.1 + NEUROPOD_PYTHON_VERSION: 3.9