mirror of
https://github.com/facebookresearch/pytorch3d.git
synced 2025-08-02 03:42:50 +08:00
00acf0b0c7
Summary: cu116 builds need to happen in a specific image. Reviewed By: patricklabatut Differential Revision: D37680352 fbshipit-source-id: 81bef0642ad832e83e4eba6321287759b3229303
739 lines
25 KiB
YAML
739 lines
25 KiB
YAML
version: 2.1
|
|
|
|
#examples:
|
|
#https://github.com/facebookresearch/ParlAI/blob/master/.circleci/config.yml
|
|
#https://github.com/facebookresearch/hydra/blob/master/.circleci/config.yml
|
|
#https://github.com/facebookresearch/habitat-api/blob/master/.circleci/config.yml
|
|
|
|
#drive tests with nox or tox or pytest?
|
|
|
|
# -------------------------------------------------------------------------------------
|
|
# environments where we run our jobs
|
|
# -------------------------------------------------------------------------------------
|
|
|
|
|
|
setupcuda: &setupcuda
|
|
run:
|
|
name: Setup CUDA
|
|
working_directory: ~/
|
|
command: |
|
|
# download and install nvidia drivers, cuda, etc
|
|
wget --no-verbose --no-clobber -P ~/nvidia-downloads https://developer.download.nvidia.com/compute/cuda/11.3.1/local_installers/cuda_11.3.1_465.19.01_linux.run
|
|
sudo sh ~/nvidia-downloads/cuda_11.3.1_465.19.01_linux.run --silent
|
|
echo "Done installing CUDA."
|
|
pyenv versions
|
|
nvidia-smi
|
|
pyenv global 3.9.1
|
|
|
|
binary_common: &binary_common
|
|
parameters:
|
|
# Edit these defaults to do a release`
|
|
build_version:
|
|
description: "version number of release binary; by default, build a nightly"
|
|
type: string
|
|
default: ""
|
|
pytorch_version:
|
|
description: "PyTorch version to build against; by default, use a nightly"
|
|
type: string
|
|
default: ""
|
|
# Don't edit these
|
|
python_version:
|
|
description: "Python version to build against (e.g., 3.7)"
|
|
type: string
|
|
cu_version:
|
|
description: "CUDA version to build against, in CU format (e.g., cpu or cu100)"
|
|
type: string
|
|
wheel_docker_image:
|
|
description: "Wheel only: what docker image to use"
|
|
type: string
|
|
default: "pytorch/manylinux-cuda101"
|
|
conda_docker_image:
|
|
description: "what docker image to use for docker"
|
|
type: string
|
|
default: "pytorch/conda-cuda"
|
|
environment:
|
|
PYTHON_VERSION: << parameters.python_version >>
|
|
BUILD_VERSION: << parameters.build_version >>
|
|
PYTORCH_VERSION: << parameters.pytorch_version >>
|
|
CU_VERSION: << parameters.cu_version >>
|
|
TESTRUN_DOCKER_IMAGE: << parameters.conda_docker_image >>
|
|
|
|
jobs:
|
|
main:
|
|
environment:
|
|
CUDA_VERSION: "11.3"
|
|
resource_class: gpu.nvidia.small.multi
|
|
machine:
|
|
image: ubuntu-2004:202101-01
|
|
steps:
|
|
- checkout
|
|
- <<: *setupcuda
|
|
- run: pip3 install --progress-bar off imageio wheel matplotlib 'pillow<7'
|
|
- run: pip3 install --progress-bar off torch==1.10.0+cu113 torchvision==0.11.1+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
|
|
# - run: conda create -p ~/conda_env python=3.7 numpy
|
|
# - run: conda activate ~/conda_env
|
|
# - run: conda install -c pytorch pytorch torchvision
|
|
|
|
- run: pip3 install --progress-bar off 'git+https://github.com/facebookresearch/fvcore'
|
|
- run: pip3 install --progress-bar off 'git+https://github.com/facebookresearch/iopath'
|
|
- run:
|
|
name: build
|
|
command: |
|
|
export LD_LIBRARY_PATH=$LD_LIBARY_PATH:/usr/local/cuda-11.3/lib64
|
|
python3 setup.py build_ext --inplace
|
|
- run: LD_LIBRARY_PATH=$LD_LIBARY_PATH:/usr/local/cuda-11.3/lib64 python -m unittest discover -v -s tests -t .
|
|
- run: python3 setup.py bdist_wheel
|
|
|
|
binary_linux_wheel:
|
|
<<: *binary_common
|
|
docker:
|
|
- image: << parameters.wheel_docker_image >>
|
|
auth:
|
|
username: $DOCKERHUB_USERNAME
|
|
password: $DOCKERHUB_TOKEN
|
|
resource_class: 2xlarge+
|
|
steps:
|
|
- checkout
|
|
- run: MAX_JOBS=15 packaging/build_wheel.sh
|
|
- store_artifacts:
|
|
path: dist
|
|
- persist_to_workspace:
|
|
root: dist
|
|
paths:
|
|
- "*"
|
|
|
|
binary_linux_conda:
|
|
<<: *binary_common
|
|
docker:
|
|
- image: "<< parameters.conda_docker_image >>"
|
|
auth:
|
|
username: $DOCKERHUB_USERNAME
|
|
password: $DOCKERHUB_TOKEN
|
|
resource_class: 2xlarge+
|
|
steps:
|
|
- checkout
|
|
# This is building with cuda but no gpu present,
|
|
# so we aren't running the tests.
|
|
- run:
|
|
name: build
|
|
no_output_timeout: 20m
|
|
command: MAX_JOBS=15 TEST_FLAG=--no-test packaging/build_conda.sh
|
|
- store_artifacts:
|
|
path: /opt/conda/conda-bld/linux-64
|
|
- persist_to_workspace:
|
|
root: /opt/conda/conda-bld/linux-64
|
|
paths:
|
|
- "*"
|
|
|
|
binary_linux_conda_cuda:
|
|
<<: *binary_common
|
|
machine:
|
|
image: ubuntu-1604-cuda-10.2:202012-01
|
|
resource_class: gpu.nvidia.small.multi
|
|
steps:
|
|
- checkout
|
|
|
|
- run:
|
|
name: Pull docker image
|
|
command: |
|
|
nvidia-smi
|
|
set -e
|
|
|
|
{ docker login -u="$DOCKERHUB_USERNAME" -p="$DOCKERHUB_TOKEN" ; } 2> /dev/null
|
|
|
|
echo Pulling docker image $TESTRUN_DOCKER_IMAGE
|
|
docker pull $TESTRUN_DOCKER_IMAGE
|
|
- run:
|
|
name: Build and run tests
|
|
no_output_timeout: 20m
|
|
command: |
|
|
set -e
|
|
|
|
cd ${HOME}/project/
|
|
|
|
export JUST_TESTRUN=1
|
|
VARS_TO_PASS="-e PYTHON_VERSION -e BUILD_VERSION -e PYTORCH_VERSION -e CU_VERSION -e JUST_TESTRUN"
|
|
|
|
docker run --gpus all --ipc=host -v $(pwd):/remote -w /remote ${VARS_TO_PASS} ${TESTRUN_DOCKER_IMAGE} ./packaging/build_conda.sh
|
|
|
|
binary_macos_wheel:
|
|
<<: *binary_common
|
|
macos:
|
|
xcode: "12.0"
|
|
steps:
|
|
- checkout
|
|
- run:
|
|
# Cannot easily deduplicate this as source'ing activate
|
|
# will set environment variables which we need to propagate
|
|
# to build_wheel.sh
|
|
command: |
|
|
curl -o conda.sh https://repo.anaconda.com/miniconda/Miniconda3-latest-MacOSX-x86_64.sh
|
|
sh conda.sh -b
|
|
source $HOME/miniconda3/bin/activate
|
|
packaging/build_wheel.sh
|
|
- store_artifacts:
|
|
path: dist
|
|
|
|
workflows:
|
|
version: 2
|
|
build_and_test:
|
|
jobs:
|
|
# - main:
|
|
# context: DOCKERHUB_TOKEN
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu101
|
|
name: linux_conda_py37_cu101_pyt180
|
|
python_version: '3.7'
|
|
pytorch_version: 1.8.0
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu102
|
|
name: linux_conda_py37_cu102_pyt180
|
|
python_version: '3.7'
|
|
pytorch_version: 1.8.0
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu111
|
|
name: linux_conda_py37_cu111_pyt180
|
|
python_version: '3.7'
|
|
pytorch_version: 1.8.0
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu101
|
|
name: linux_conda_py37_cu101_pyt181
|
|
python_version: '3.7'
|
|
pytorch_version: 1.8.1
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu102
|
|
name: linux_conda_py37_cu102_pyt181
|
|
python_version: '3.7'
|
|
pytorch_version: 1.8.1
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu111
|
|
name: linux_conda_py37_cu111_pyt181
|
|
python_version: '3.7'
|
|
pytorch_version: 1.8.1
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu102
|
|
name: linux_conda_py37_cu102_pyt190
|
|
python_version: '3.7'
|
|
pytorch_version: 1.9.0
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu111
|
|
name: linux_conda_py37_cu111_pyt190
|
|
python_version: '3.7'
|
|
pytorch_version: 1.9.0
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu102
|
|
name: linux_conda_py37_cu102_pyt191
|
|
python_version: '3.7'
|
|
pytorch_version: 1.9.1
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu111
|
|
name: linux_conda_py37_cu111_pyt191
|
|
python_version: '3.7'
|
|
pytorch_version: 1.9.1
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu102
|
|
name: linux_conda_py37_cu102_pyt1100
|
|
python_version: '3.7'
|
|
pytorch_version: 1.10.0
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu111
|
|
name: linux_conda_py37_cu111_pyt1100
|
|
python_version: '3.7'
|
|
pytorch_version: 1.10.0
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda113
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu113
|
|
name: linux_conda_py37_cu113_pyt1100
|
|
python_version: '3.7'
|
|
pytorch_version: 1.10.0
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu102
|
|
name: linux_conda_py37_cu102_pyt1101
|
|
python_version: '3.7'
|
|
pytorch_version: 1.10.1
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu111
|
|
name: linux_conda_py37_cu111_pyt1101
|
|
python_version: '3.7'
|
|
pytorch_version: 1.10.1
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda113
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu113
|
|
name: linux_conda_py37_cu113_pyt1101
|
|
python_version: '3.7'
|
|
pytorch_version: 1.10.1
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu102
|
|
name: linux_conda_py37_cu102_pyt1102
|
|
python_version: '3.7'
|
|
pytorch_version: 1.10.2
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu111
|
|
name: linux_conda_py37_cu111_pyt1102
|
|
python_version: '3.7'
|
|
pytorch_version: 1.10.2
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda113
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu113
|
|
name: linux_conda_py37_cu113_pyt1102
|
|
python_version: '3.7'
|
|
pytorch_version: 1.10.2
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu102
|
|
name: linux_conda_py37_cu102_pyt1110
|
|
python_version: '3.7'
|
|
pytorch_version: 1.11.0
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu111
|
|
name: linux_conda_py37_cu111_pyt1110
|
|
python_version: '3.7'
|
|
pytorch_version: 1.11.0
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda113
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu113
|
|
name: linux_conda_py37_cu113_pyt1110
|
|
python_version: '3.7'
|
|
pytorch_version: 1.11.0
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda115
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu115
|
|
name: linux_conda_py37_cu115_pyt1110
|
|
python_version: '3.7'
|
|
pytorch_version: 1.11.0
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu102
|
|
name: linux_conda_py37_cu102_pyt1120
|
|
python_version: '3.7'
|
|
pytorch_version: 1.12.0
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda113
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu113
|
|
name: linux_conda_py37_cu113_pyt1120
|
|
python_version: '3.7'
|
|
pytorch_version: 1.12.0
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda116
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu116
|
|
name: linux_conda_py37_cu116_pyt1120
|
|
python_version: '3.7'
|
|
pytorch_version: 1.12.0
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu101
|
|
name: linux_conda_py38_cu101_pyt180
|
|
python_version: '3.8'
|
|
pytorch_version: 1.8.0
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu102
|
|
name: linux_conda_py38_cu102_pyt180
|
|
python_version: '3.8'
|
|
pytorch_version: 1.8.0
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu111
|
|
name: linux_conda_py38_cu111_pyt180
|
|
python_version: '3.8'
|
|
pytorch_version: 1.8.0
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu101
|
|
name: linux_conda_py38_cu101_pyt181
|
|
python_version: '3.8'
|
|
pytorch_version: 1.8.1
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu102
|
|
name: linux_conda_py38_cu102_pyt181
|
|
python_version: '3.8'
|
|
pytorch_version: 1.8.1
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu111
|
|
name: linux_conda_py38_cu111_pyt181
|
|
python_version: '3.8'
|
|
pytorch_version: 1.8.1
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu102
|
|
name: linux_conda_py38_cu102_pyt190
|
|
python_version: '3.8'
|
|
pytorch_version: 1.9.0
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu111
|
|
name: linux_conda_py38_cu111_pyt190
|
|
python_version: '3.8'
|
|
pytorch_version: 1.9.0
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu102
|
|
name: linux_conda_py38_cu102_pyt191
|
|
python_version: '3.8'
|
|
pytorch_version: 1.9.1
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu111
|
|
name: linux_conda_py38_cu111_pyt191
|
|
python_version: '3.8'
|
|
pytorch_version: 1.9.1
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu102
|
|
name: linux_conda_py38_cu102_pyt1100
|
|
python_version: '3.8'
|
|
pytorch_version: 1.10.0
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu111
|
|
name: linux_conda_py38_cu111_pyt1100
|
|
python_version: '3.8'
|
|
pytorch_version: 1.10.0
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda113
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu113
|
|
name: linux_conda_py38_cu113_pyt1100
|
|
python_version: '3.8'
|
|
pytorch_version: 1.10.0
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu102
|
|
name: linux_conda_py38_cu102_pyt1101
|
|
python_version: '3.8'
|
|
pytorch_version: 1.10.1
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu111
|
|
name: linux_conda_py38_cu111_pyt1101
|
|
python_version: '3.8'
|
|
pytorch_version: 1.10.1
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda113
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu113
|
|
name: linux_conda_py38_cu113_pyt1101
|
|
python_version: '3.8'
|
|
pytorch_version: 1.10.1
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu102
|
|
name: linux_conda_py38_cu102_pyt1102
|
|
python_version: '3.8'
|
|
pytorch_version: 1.10.2
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu111
|
|
name: linux_conda_py38_cu111_pyt1102
|
|
python_version: '3.8'
|
|
pytorch_version: 1.10.2
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda113
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu113
|
|
name: linux_conda_py38_cu113_pyt1102
|
|
python_version: '3.8'
|
|
pytorch_version: 1.10.2
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu102
|
|
name: linux_conda_py38_cu102_pyt1110
|
|
python_version: '3.8'
|
|
pytorch_version: 1.11.0
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu111
|
|
name: linux_conda_py38_cu111_pyt1110
|
|
python_version: '3.8'
|
|
pytorch_version: 1.11.0
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda113
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu113
|
|
name: linux_conda_py38_cu113_pyt1110
|
|
python_version: '3.8'
|
|
pytorch_version: 1.11.0
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda115
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu115
|
|
name: linux_conda_py38_cu115_pyt1110
|
|
python_version: '3.8'
|
|
pytorch_version: 1.11.0
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu102
|
|
name: linux_conda_py38_cu102_pyt1120
|
|
python_version: '3.8'
|
|
pytorch_version: 1.12.0
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda113
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu113
|
|
name: linux_conda_py38_cu113_pyt1120
|
|
python_version: '3.8'
|
|
pytorch_version: 1.12.0
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda116
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu116
|
|
name: linux_conda_py38_cu116_pyt1120
|
|
python_version: '3.8'
|
|
pytorch_version: 1.12.0
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu101
|
|
name: linux_conda_py39_cu101_pyt180
|
|
python_version: '3.9'
|
|
pytorch_version: 1.8.0
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu102
|
|
name: linux_conda_py39_cu102_pyt180
|
|
python_version: '3.9'
|
|
pytorch_version: 1.8.0
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu111
|
|
name: linux_conda_py39_cu111_pyt180
|
|
python_version: '3.9'
|
|
pytorch_version: 1.8.0
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu101
|
|
name: linux_conda_py39_cu101_pyt181
|
|
python_version: '3.9'
|
|
pytorch_version: 1.8.1
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu102
|
|
name: linux_conda_py39_cu102_pyt181
|
|
python_version: '3.9'
|
|
pytorch_version: 1.8.1
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu111
|
|
name: linux_conda_py39_cu111_pyt181
|
|
python_version: '3.9'
|
|
pytorch_version: 1.8.1
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu102
|
|
name: linux_conda_py39_cu102_pyt190
|
|
python_version: '3.9'
|
|
pytorch_version: 1.9.0
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu111
|
|
name: linux_conda_py39_cu111_pyt190
|
|
python_version: '3.9'
|
|
pytorch_version: 1.9.0
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu102
|
|
name: linux_conda_py39_cu102_pyt191
|
|
python_version: '3.9'
|
|
pytorch_version: 1.9.1
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu111
|
|
name: linux_conda_py39_cu111_pyt191
|
|
python_version: '3.9'
|
|
pytorch_version: 1.9.1
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu102
|
|
name: linux_conda_py39_cu102_pyt1100
|
|
python_version: '3.9'
|
|
pytorch_version: 1.10.0
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu111
|
|
name: linux_conda_py39_cu111_pyt1100
|
|
python_version: '3.9'
|
|
pytorch_version: 1.10.0
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda113
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu113
|
|
name: linux_conda_py39_cu113_pyt1100
|
|
python_version: '3.9'
|
|
pytorch_version: 1.10.0
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu102
|
|
name: linux_conda_py39_cu102_pyt1101
|
|
python_version: '3.9'
|
|
pytorch_version: 1.10.1
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu111
|
|
name: linux_conda_py39_cu111_pyt1101
|
|
python_version: '3.9'
|
|
pytorch_version: 1.10.1
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda113
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu113
|
|
name: linux_conda_py39_cu113_pyt1101
|
|
python_version: '3.9'
|
|
pytorch_version: 1.10.1
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu102
|
|
name: linux_conda_py39_cu102_pyt1102
|
|
python_version: '3.9'
|
|
pytorch_version: 1.10.2
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu111
|
|
name: linux_conda_py39_cu111_pyt1102
|
|
python_version: '3.9'
|
|
pytorch_version: 1.10.2
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda113
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu113
|
|
name: linux_conda_py39_cu113_pyt1102
|
|
python_version: '3.9'
|
|
pytorch_version: 1.10.2
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu102
|
|
name: linux_conda_py39_cu102_pyt1110
|
|
python_version: '3.9'
|
|
pytorch_version: 1.11.0
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu111
|
|
name: linux_conda_py39_cu111_pyt1110
|
|
python_version: '3.9'
|
|
pytorch_version: 1.11.0
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda113
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu113
|
|
name: linux_conda_py39_cu113_pyt1110
|
|
python_version: '3.9'
|
|
pytorch_version: 1.11.0
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda115
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu115
|
|
name: linux_conda_py39_cu115_pyt1110
|
|
python_version: '3.9'
|
|
pytorch_version: 1.11.0
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu102
|
|
name: linux_conda_py39_cu102_pyt1120
|
|
python_version: '3.9'
|
|
pytorch_version: 1.12.0
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda113
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu113
|
|
name: linux_conda_py39_cu113_pyt1120
|
|
python_version: '3.9'
|
|
pytorch_version: 1.12.0
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda116
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu116
|
|
name: linux_conda_py39_cu116_pyt1120
|
|
python_version: '3.9'
|
|
pytorch_version: 1.12.0
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu102
|
|
name: linux_conda_py310_cu102_pyt1110
|
|
python_version: '3.10'
|
|
pytorch_version: 1.11.0
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu111
|
|
name: linux_conda_py310_cu111_pyt1110
|
|
python_version: '3.10'
|
|
pytorch_version: 1.11.0
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda113
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu113
|
|
name: linux_conda_py310_cu113_pyt1110
|
|
python_version: '3.10'
|
|
pytorch_version: 1.11.0
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda115
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu115
|
|
name: linux_conda_py310_cu115_pyt1110
|
|
python_version: '3.10'
|
|
pytorch_version: 1.11.0
|
|
- binary_linux_conda:
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu102
|
|
name: linux_conda_py310_cu102_pyt1120
|
|
python_version: '3.10'
|
|
pytorch_version: 1.12.0
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda113
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu113
|
|
name: linux_conda_py310_cu113_pyt1120
|
|
python_version: '3.10'
|
|
pytorch_version: 1.12.0
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda116
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu116
|
|
name: linux_conda_py310_cu116_pyt1120
|
|
python_version: '3.10'
|
|
pytorch_version: 1.12.0
|
|
- binary_linux_conda_cuda:
|
|
name: testrun_conda_cuda_py37_cu102_pyt190
|
|
context: DOCKERHUB_TOKEN
|
|
python_version: "3.7"
|
|
pytorch_version: '1.9.0'
|
|
cu_version: "cu102"
|
|
- binary_macos_wheel:
|
|
cu_version: cpu
|
|
name: macos_wheel_py37_cpu
|
|
python_version: '3.7'
|
|
pytorch_version: '1.12.0'
|
|
- binary_macos_wheel:
|
|
cu_version: cpu
|
|
name: macos_wheel_py38_cpu
|
|
python_version: '3.8'
|
|
pytorch_version: '1.12.0'
|
|
- binary_macos_wheel:
|
|
cu_version: cpu
|
|
name: macos_wheel_py39_cpu
|
|
python_version: '3.9'
|
|
pytorch_version: '1.12.0'
|