mirror of
https://github.com/facebookresearch/pytorch3d.git
synced 2025-07-31 10:52:50 +08:00
d0d9cae9cd
Summary: Build for new pytorch versions Reviewed By: shapovalov Differential Revision: D53266104 fbshipit-source-id: f7aaacaf39cab3839b24f45361c36f087d0ea7c9
661 lines
24 KiB
YAML
661 lines
24 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: linux-cuda-11:default
|
|
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: 40m
|
|
command: MAX_JOBS=15 TEST_FLAG=--no-test python3 packaging/build_conda.py
|
|
- 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: linux-cuda-11:default
|
|
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: 40m
|
|
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} python3 ./packaging/build_conda.py
|
|
|
|
workflows:
|
|
version: 2
|
|
build_and_test:
|
|
jobs:
|
|
# - main:
|
|
# context: DOCKERHUB_TOKEN
|
|
- 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:
|
|
conda_docker_image: pytorch/conda-builder:cuda113
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu113
|
|
name: linux_conda_py38_cu113_pyt1121
|
|
python_version: '3.8'
|
|
pytorch_version: 1.12.1
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda116
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu116
|
|
name: linux_conda_py38_cu116_pyt1121
|
|
python_version: '3.8'
|
|
pytorch_version: 1.12.1
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda116
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu116
|
|
name: linux_conda_py38_cu116_pyt1130
|
|
python_version: '3.8'
|
|
pytorch_version: 1.13.0
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda117
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu117
|
|
name: linux_conda_py38_cu117_pyt1130
|
|
python_version: '3.8'
|
|
pytorch_version: 1.13.0
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda116
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu116
|
|
name: linux_conda_py38_cu116_pyt1131
|
|
python_version: '3.8'
|
|
pytorch_version: 1.13.1
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda117
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu117
|
|
name: linux_conda_py38_cu117_pyt1131
|
|
python_version: '3.8'
|
|
pytorch_version: 1.13.1
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda117
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu117
|
|
name: linux_conda_py38_cu117_pyt200
|
|
python_version: '3.8'
|
|
pytorch_version: 2.0.0
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda118
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu118
|
|
name: linux_conda_py38_cu118_pyt200
|
|
python_version: '3.8'
|
|
pytorch_version: 2.0.0
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda117
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu117
|
|
name: linux_conda_py38_cu117_pyt201
|
|
python_version: '3.8'
|
|
pytorch_version: 2.0.1
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda118
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu118
|
|
name: linux_conda_py38_cu118_pyt201
|
|
python_version: '3.8'
|
|
pytorch_version: 2.0.1
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda118
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu118
|
|
name: linux_conda_py38_cu118_pyt210
|
|
python_version: '3.8'
|
|
pytorch_version: 2.1.0
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda121
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu121
|
|
name: linux_conda_py38_cu121_pyt210
|
|
python_version: '3.8'
|
|
pytorch_version: 2.1.0
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda118
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu118
|
|
name: linux_conda_py38_cu118_pyt211
|
|
python_version: '3.8'
|
|
pytorch_version: 2.1.1
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda121
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu121
|
|
name: linux_conda_py38_cu121_pyt211
|
|
python_version: '3.8'
|
|
pytorch_version: 2.1.1
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda118
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu118
|
|
name: linux_conda_py38_cu118_pyt212
|
|
python_version: '3.8'
|
|
pytorch_version: 2.1.2
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda121
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu121
|
|
name: linux_conda_py38_cu121_pyt212
|
|
python_version: '3.8'
|
|
pytorch_version: 2.1.2
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda118
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu118
|
|
name: linux_conda_py38_cu118_pyt220
|
|
python_version: '3.8'
|
|
pytorch_version: 2.2.0
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda121
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu121
|
|
name: linux_conda_py38_cu121_pyt220
|
|
python_version: '3.8'
|
|
pytorch_version: 2.2.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:
|
|
conda_docker_image: pytorch/conda-builder:cuda113
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu113
|
|
name: linux_conda_py39_cu113_pyt1121
|
|
python_version: '3.9'
|
|
pytorch_version: 1.12.1
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda116
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu116
|
|
name: linux_conda_py39_cu116_pyt1121
|
|
python_version: '3.9'
|
|
pytorch_version: 1.12.1
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda116
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu116
|
|
name: linux_conda_py39_cu116_pyt1130
|
|
python_version: '3.9'
|
|
pytorch_version: 1.13.0
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda117
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu117
|
|
name: linux_conda_py39_cu117_pyt1130
|
|
python_version: '3.9'
|
|
pytorch_version: 1.13.0
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda116
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu116
|
|
name: linux_conda_py39_cu116_pyt1131
|
|
python_version: '3.9'
|
|
pytorch_version: 1.13.1
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda117
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu117
|
|
name: linux_conda_py39_cu117_pyt1131
|
|
python_version: '3.9'
|
|
pytorch_version: 1.13.1
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda117
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu117
|
|
name: linux_conda_py39_cu117_pyt200
|
|
python_version: '3.9'
|
|
pytorch_version: 2.0.0
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda118
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu118
|
|
name: linux_conda_py39_cu118_pyt200
|
|
python_version: '3.9'
|
|
pytorch_version: 2.0.0
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda117
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu117
|
|
name: linux_conda_py39_cu117_pyt201
|
|
python_version: '3.9'
|
|
pytorch_version: 2.0.1
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda118
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu118
|
|
name: linux_conda_py39_cu118_pyt201
|
|
python_version: '3.9'
|
|
pytorch_version: 2.0.1
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda118
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu118
|
|
name: linux_conda_py39_cu118_pyt210
|
|
python_version: '3.9'
|
|
pytorch_version: 2.1.0
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda121
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu121
|
|
name: linux_conda_py39_cu121_pyt210
|
|
python_version: '3.9'
|
|
pytorch_version: 2.1.0
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda118
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu118
|
|
name: linux_conda_py39_cu118_pyt211
|
|
python_version: '3.9'
|
|
pytorch_version: 2.1.1
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda121
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu121
|
|
name: linux_conda_py39_cu121_pyt211
|
|
python_version: '3.9'
|
|
pytorch_version: 2.1.1
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda118
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu118
|
|
name: linux_conda_py39_cu118_pyt212
|
|
python_version: '3.9'
|
|
pytorch_version: 2.1.2
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda121
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu121
|
|
name: linux_conda_py39_cu121_pyt212
|
|
python_version: '3.9'
|
|
pytorch_version: 2.1.2
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda118
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu118
|
|
name: linux_conda_py39_cu118_pyt220
|
|
python_version: '3.9'
|
|
pytorch_version: 2.2.0
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda121
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu121
|
|
name: linux_conda_py39_cu121_pyt220
|
|
python_version: '3.9'
|
|
pytorch_version: 2.2.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:
|
|
conda_docker_image: pytorch/conda-builder:cuda113
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu113
|
|
name: linux_conda_py310_cu113_pyt1121
|
|
python_version: '3.10'
|
|
pytorch_version: 1.12.1
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda116
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu116
|
|
name: linux_conda_py310_cu116_pyt1121
|
|
python_version: '3.10'
|
|
pytorch_version: 1.12.1
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda116
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu116
|
|
name: linux_conda_py310_cu116_pyt1130
|
|
python_version: '3.10'
|
|
pytorch_version: 1.13.0
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda117
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu117
|
|
name: linux_conda_py310_cu117_pyt1130
|
|
python_version: '3.10'
|
|
pytorch_version: 1.13.0
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda116
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu116
|
|
name: linux_conda_py310_cu116_pyt1131
|
|
python_version: '3.10'
|
|
pytorch_version: 1.13.1
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda117
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu117
|
|
name: linux_conda_py310_cu117_pyt1131
|
|
python_version: '3.10'
|
|
pytorch_version: 1.13.1
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda117
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu117
|
|
name: linux_conda_py310_cu117_pyt200
|
|
python_version: '3.10'
|
|
pytorch_version: 2.0.0
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda118
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu118
|
|
name: linux_conda_py310_cu118_pyt200
|
|
python_version: '3.10'
|
|
pytorch_version: 2.0.0
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda117
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu117
|
|
name: linux_conda_py310_cu117_pyt201
|
|
python_version: '3.10'
|
|
pytorch_version: 2.0.1
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda118
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu118
|
|
name: linux_conda_py310_cu118_pyt201
|
|
python_version: '3.10'
|
|
pytorch_version: 2.0.1
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda118
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu118
|
|
name: linux_conda_py310_cu118_pyt210
|
|
python_version: '3.10'
|
|
pytorch_version: 2.1.0
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda121
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu121
|
|
name: linux_conda_py310_cu121_pyt210
|
|
python_version: '3.10'
|
|
pytorch_version: 2.1.0
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda118
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu118
|
|
name: linux_conda_py310_cu118_pyt211
|
|
python_version: '3.10'
|
|
pytorch_version: 2.1.1
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda121
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu121
|
|
name: linux_conda_py310_cu121_pyt211
|
|
python_version: '3.10'
|
|
pytorch_version: 2.1.1
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda118
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu118
|
|
name: linux_conda_py310_cu118_pyt212
|
|
python_version: '3.10'
|
|
pytorch_version: 2.1.2
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda121
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu121
|
|
name: linux_conda_py310_cu121_pyt212
|
|
python_version: '3.10'
|
|
pytorch_version: 2.1.2
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda118
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu118
|
|
name: linux_conda_py310_cu118_pyt220
|
|
python_version: '3.10'
|
|
pytorch_version: 2.2.0
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda121
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu121
|
|
name: linux_conda_py310_cu121_pyt220
|
|
python_version: '3.10'
|
|
pytorch_version: 2.2.0
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda118
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu118
|
|
name: linux_conda_py311_cu118_pyt210
|
|
python_version: '3.11'
|
|
pytorch_version: 2.1.0
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda121
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu121
|
|
name: linux_conda_py311_cu121_pyt210
|
|
python_version: '3.11'
|
|
pytorch_version: 2.1.0
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda118
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu118
|
|
name: linux_conda_py311_cu118_pyt211
|
|
python_version: '3.11'
|
|
pytorch_version: 2.1.1
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda121
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu121
|
|
name: linux_conda_py311_cu121_pyt211
|
|
python_version: '3.11'
|
|
pytorch_version: 2.1.1
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda118
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu118
|
|
name: linux_conda_py311_cu118_pyt212
|
|
python_version: '3.11'
|
|
pytorch_version: 2.1.2
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda121
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu121
|
|
name: linux_conda_py311_cu121_pyt212
|
|
python_version: '3.11'
|
|
pytorch_version: 2.1.2
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda118
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu118
|
|
name: linux_conda_py311_cu118_pyt220
|
|
python_version: '3.11'
|
|
pytorch_version: 2.2.0
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda121
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu121
|
|
name: linux_conda_py311_cu121_pyt220
|
|
python_version: '3.11'
|
|
pytorch_version: 2.2.0
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda118
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu118
|
|
name: linux_conda_py312_cu118_pyt220
|
|
python_version: '3.12'
|
|
pytorch_version: 2.2.0
|
|
- binary_linux_conda:
|
|
conda_docker_image: pytorch/conda-builder:cuda121
|
|
context: DOCKERHUB_TOKEN
|
|
cu_version: cu121
|
|
name: linux_conda_py312_cu121_pyt220
|
|
python_version: '3.12'
|
|
pytorch_version: 2.2.0
|
|
- binary_linux_conda_cuda:
|
|
name: testrun_conda_cuda_py310_cu117_pyt201
|
|
context: DOCKERHUB_TOKEN
|
|
python_version: "3.10"
|
|
pytorch_version: '2.0.1'
|
|
cu_version: "cu117"
|