mirror of
https://github.com/facebookresearch/pytorch3d.git
synced 2025-08-23 22:32:51 +08:00
0fecb2ddb9
Summary: Pytorch 1.5 is coming soon. I imagine we will want the ability to upload conda packages for pytorch3d to anaconda cloud for each of pytorch 1.4 and pytorch 1.5. This change adds the dependent pytorch version to the name of the conda package to make that feasible. As an example, a built package after this change will have a name like `linux-64/pytorch3d-0.1.1-py38_cu100_pyt14.tar.bz2`, instead of simply `linux-64/pytorch3d-0.1.1-py38_cu100.tar.bz2`. Also some tiny cleanup of circleci config. Other alternatives: (1) forcing users to update pytorch and pytorch3d together, (2) trying to get away with one build for multiple pytorch versions. Reviewed By: nikhilaravi Differential Revision: D20599039 fbshipit-source-id: 20164eda4a5141afed47b3596e559950d796ffc9
61 lines
1.2 KiB
YAML
61 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]
|
|
|
|
host:
|
|
- python
|
|
- setuptools
|
|
{{ environ.get('CONDA_PYTORCH_BUILD_CONSTRAINT') }}
|
|
{{ environ.get('CONDA_CUDATOOLKIT_CONSTRAINT') }}
|
|
{{ environ.get('CONDA_CPUONLY_FEATURE') }}
|
|
|
|
run:
|
|
- python
|
|
- numpy >=1.11
|
|
- six
|
|
- torchvision >=0.5
|
|
- fvcore
|
|
{{ 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
|
|
features:
|
|
{{ environ.get('CONDA_CPUONLY_FEATURE') }}
|
|
|
|
test:
|
|
imports:
|
|
- pytorch3d
|
|
source_files:
|
|
- tests
|
|
- docs
|
|
requires:
|
|
- pytest
|
|
- scipy
|
|
- mock
|
|
- av
|
|
- ca-certificates
|
|
- typing
|
|
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'
|