blob: a93703c3eaa5a86ad91a50147c2eacc5a3677e0b (
plain) (
blame)
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
|
PORTNAME= torchvision
DISTVERSIONPREFIX= v
DISTVERSION= 0.22.0
CATEGORIES= misc # machine-learning
PKGNAMEPREFIX= ${PYTHON_PKGNAMEPREFIX}
MAINTAINER= yuri@FreeBSD.org
COMMENT= PyTorch: Datasets, transforms and models specific to computer vision
WWW= https://pytorch.org/vision/stable/index.html \
https://github.com/pytorch/vision
LICENSE= BSD3CLAUSE
LICENSE_FILE= ${WRKSRC}/LICENSE
LIB_DEPENDS= libpng16.so:graphics/png \
libwebp.so:graphics/webp
PY_DEPENDS= ${PYNUMPY} \
${PYTHON_PKGNAMEPREFIX}requests>0:www/py-requests@${PY_FLAVOR} \
${PYTHON_PKGNAMEPREFIX}pytorch>0:misc/py-pytorch@${PY_FLAVOR}
PY_DEPENDS+= ${PYTHON_PKGNAMEPREFIX}pillow>=5.3.0:graphics/py-pillow@${PY_FLAVOR} # image backend
BUILD_DEPENDS= ${PY_SETUPTOOLS} \
${PY_DEPENDS} \
${PYTHON_PKGNAMEPREFIX}fsspec>0:filesystems/py-fsspec@${PY_FLAVOR} \
pybind11>0:devel/pybind11 \
${PYTHON_PKGNAMEPREFIX}wheel>0:devel/py-wheel@${PY_FLAVOR}
BUILD_DEPENDS+= ninja:devel/ninja # build uses the C compiler for C++ files w/out ninja, and fails
RUN_DEPENDS= ${PY_DEPENDS}
USES= compiler:c++17-lang jpeg python
USE_PYTHON= pep517 autoplist pytest # tests run in a very high memory ; in 0.17.1 tests fail to run, see https://github.com/pytorch/vision/issues/8278
USE_GITHUB= yes
GH_ACCOUNT= pytorch
GH_PROJECT= vision
MAKE_ENV= TORCHVISION_INCLUDE=${LOCALBASE}/include # workaround for build failure suggested here: https://github.com/pytorch/vision/issues/8397#issuecomment-2168351425
post-install: # strip binaries
@${STRIP_CMD} \
${STAGEDIR}${PYTHON_SITELIBDIR}/torchvision/_C.so \
${STAGEDIR}${PYTHON_SITELIBDIR}/torchvision/image.so
.include <bsd.port.mk>
|
