-
Notifications
You must be signed in to change notification settings - Fork 6.9k
/
meta.yaml
74 lines (65 loc) · 1.71 KB
/
meta.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
{% set build_variant = environ.get('CONDA_BUILD_VARIANT', 'cpu') %}
package:
name: torchvision
version: "{{ environ.get('BUILD_VERSION') }}"
source:
path: "{{ environ.get('SOURCE_ROOT_DIR') }}"
requirements:
build:
- {{ compiler('c') }} # [win]
- libpng
- libjpeg-turbo
- libwebp
- ffmpeg >=4.2.2, <5.0.0 # [linux]
host:
- python
- setuptools
- pytorch-mutex 1.0 {{ build_variant }} # [not osx ]
{{ environ.get('CONDA_PYTORCH_BUILD_CONSTRAINT', 'pytorch') }}
{{ environ.get('CONDA_CUDATOOLKIT_CONSTRAINT', '') }}
run:
- python
- defaults::numpy >=1.11 # [py <= 310]
- numpy >=1.23.5 # [py >= 311]
- requests
- libpng
- ffmpeg >=4.2.2, <5.0.0 # [linux]
- libjpeg-turbo
- libwebp
- pillow >=5.3.0, !=8.3.*
- pytorch-mutex 1.0 {{ build_variant }} # [not osx ]
{{ environ.get('CONDA_PYTORCH_CONSTRAINT', 'pytorch') }}
{{ environ.get('CONDA_CUDATOOLKIT_CONSTRAINT', '') }}
{% if build_variant == 'cpu' %}
run_constrained:
- cpuonly
{% elif not osx %}
run_constrained:
- cpuonly <0
{% endif %}
build:
string: py{{py}}_{{ environ['CU_VERSION'] }}
script: python setup.py install --single-version-externally-managed --record=record.txt
script_env:
- CUDA_HOME
- FORCE_CUDA
- BUILD_VERSION
- TORCH_CUDA_ARCH_LIST
- MACOSX_DEPLOYMENT_TARGET
test:
imports:
- torchvision
- torchvision.datasets
- torchvision.transforms
source_files:
- test
requires:
- pytest
- scipy
- libjpeg-turbo
- ca-certificates
about:
home: https://github.com/pytorch/vision
license: BSD
license_file: LICENSE
summary: 'image and video datasets and models for torch deep learning'