mirror of
https://github.com/facebookresearch/pytorch3d.git
synced 2025-08-02 03:42:50 +08:00
513a6476bc
Summary: Add ioPath as a dependency of PyTorch3D in preparation for using the new PathManager. Reviewed By: nikhilaravi Differential Revision: D25372971 fbshipit-source-id: d8aa661d2de975e747dd494edc42bf843990cf68
58 lines
1.2 KiB
YAML
58 lines
1.2 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
|
|
commands:
|
|
#pytest .
|
|
python -m unittest discover -v -s tests
|
|
|
|
|
|
about:
|
|
home: https://github.com/facebookresearch/pytorch3d
|
|
license: BSD
|
|
license_file: LICENSE
|
|
summary: '3d Geometry for pytorch'
|