| # WARNING: DO NOT EDIT THIS FILE DIRECTLY!!! |
| # See the README.md in this directory. |
| |
| # IMPORTANT: To update Docker image version, please follow |
| # the instructions at |
| # https://github.com/pytorch/pytorch/wiki/Docker-image-build-on-CircleCI |
| |
| version: 2.1 |
| |
| parameters: |
| run_binary_tests: |
| type: boolean |
| default: false |
| |
| docker_config_defaults: &docker_config_defaults |
| user: jenkins |
| aws_auth: |
| # This IAM user only allows read-write access to ECR |
| aws_access_key_id: ${CIRCLECI_AWS_ACCESS_KEY_FOR_ECR_READ_WRITE_V4} |
| aws_secret_access_key: ${CIRCLECI_AWS_SECRET_KEY_FOR_ECR_READ_WRITE_V4} |
| |
| executors: |
| windows-with-nvidia-gpu: |
| machine: |
| resource_class: windows.gpu.nvidia.medium |
| image: windows-server-2019-nvidia:stable |
| shell: bash.exe |
| |
| windows-xlarge-cpu-with-nvidia-cuda: |
| machine: |
| resource_class: windows.xlarge |
| image: windows-server-2019-vs2019:stable |
| shell: bash.exe |
| |
| windows-medium-cpu-with-nvidia-cuda: |
| machine: |
| resource_class: windows.medium |
| image: windows-server-2019-vs2019:stable |
| shell: bash.exe |
