mirror of
https://github.com/facebookresearch/pytorch3d.git
synced 2025-08-03 20:32:51 +08:00
faf0885811
Summary: cuda 10.2 location on linux. Also remove unused conda test dependencies. Reviewed By: nikhilaravi Differential Revision: D21176409 fbshipit-source-id: dd3f339a92233ff16877ba76506ddf8f4418715d
55 lines
1.2 KiB
YAML
55 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
|
|
- 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:
|
|
- ca-certificates
|
|
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'
|