meta.yaml 1.41 KB
Newer Older
1
package:
2
3
  name: torchvision
  version: "{{ environ.get('BUILD_VERSION') }}"
4
5

source:
6
 path: "{{ environ.get('SOURCE_ROOT_DIR') }}"
7
8
9
10

requirements:
  build:
    - {{ compiler('c') }} # [win]
11
    - libpng
12
13
14
    - jpeg <=9b
    # NOTE: The only ffmpeg version that we build is actually 4.2
    - ffmpeg >=4.2  # [not win]
15
16
17
18

  host:
    - python
    - setuptools
19
    - defaults::numpy >=1.11
20
21
    {{ environ.get('CONDA_PYTORCH_BUILD_CONSTRAINT') }}
    {{ environ.get('CONDA_CUDATOOLKIT_CONSTRAINT') }}
peterjc123's avatar
peterjc123 committed
22
    {{ environ.get('CONDA_CPUONLY_FEATURE') }}
23
24
25

  run:
    - python
26
    - libpng
27
28
    - ffmpeg >=4.2  # [not win]
    - jpeg <=9b
29
    - pillow >=4.1.1
30
    - defaults::numpy >=1.11
31
32
    {{ environ.get('CONDA_PYTORCH_CONSTRAINT') }}
    {{ environ.get('CONDA_CUDATOOLKIT_CONSTRAINT') }}
33
34

build:
35
  string: py{{py}}_{{ environ['CU_VERSION'] }}
peterjc123's avatar
peterjc123 committed
36
  script: python setup.py install --single-version-externally-managed --record=record.txt
37
  script_env:
Edward Z. Yang's avatar
Edward Z. Yang committed
38
39
40
    - CUDA_HOME
    - FORCE_CUDA
    - NVCC_FLAGS
41
    - BUILD_VERSION
42
  features:
peterjc123's avatar
peterjc123 committed
43
    {{ environ.get('CONDA_CPUONLY_FEATURE') }}
44
45
46
47
48
49
50
51
52
53
54

test:
  imports:
    - torchvision
    - torchvision.datasets
    - torchvision.transforms
  source_files:
    - test
  requires:
    - pytest
    - scipy
55
56
    - av >=8.0.1
    - jpeg <=9b
57
    - ca-certificates
58
59
60
61
62
63
64


about:
  home: https://github.com/pytorch/vision
  license: BSD
  license_file: LICENSE
  summary: 'image and video datasets and models for torch deep learning'