Jeremy Reizenstein da9584357e circleci fixes
Summary:
Misc fixes.

- most important: the mac image is gone so switch to a newer one.
- torch.concat is new; was used accidentally
- remove lpips from testing in meta.yaml as it is breaking the conda test. Better to leave the relevant tests failing in OSS.
- TypedDict usage is breaking implicitron on Python 3.7.

Reviewed By: patricklabatut

Differential Revision: D38458164

fbshipit-source-id: b16c26453a743b9a771e2a6787b9a4d2a52e41c2
2022-08-05 08:58:17 -07:00

60 lines
1.3 KiB
YAML

package:
name: pytorch3d
version: "{{ environ.get('BUILD_VERSION') }}"
source:
path: "{{ environ.get('SOURCE_ROOT_DIR') }}"
requirements:
build:
- {{ compiler('c') }} # [win]
{{ environ.get('CONDA_CUB_CONSTRAINT') }}
host:
- python
- setuptools
{{ environ.get('CONDA_PYTORCH_BUILD_CONSTRAINT') }}
{{ environ.get('CONDA_CUDATOOLKIT_CONSTRAINT') }}
{{ environ.get('CONDA_CPUONLY_FEATURE') }}
run:
- python
- numpy >=1.11
- torchvision >=0.5
- fvcore
- iopath
{{ environ.get('CONDA_PYTORCH_CONSTRAINT') }}
{{ environ.get('CONDA_CUDATOOLKIT_CONSTRAINT') }}
build:
string: py{{py}}_{{ environ['CU_VERSION'] }}_pyt{{ environ['PYTORCH_VERSION_NODOT']}}
script: python setup.py install --single-version-externally-managed --record=record.txt # [not win]
script_env:
- CUDA_HOME
- FORCE_CUDA
- NVCC_FLAGS
- MAX_JOBS
features:
{{ environ.get('CONDA_CPUONLY_FEATURE') }}
test:
imports:
- pytorch3d
source_files:
- tests
- docs
requires:
- imageio
- hydra-core
- accelerate
commands:
#pytest .
python -m unittest discover -v -s tests -t .
about:
home: https://github.com/facebookresearch/pytorch3d
license: BSD
license_file: LICENSE
summary: '3d Geometry for pytorch'