mirror of
https://github.com/termux-pacman/termux-packages.git
synced 2026-02-10 20:00:51 +00:00
36 lines
1.4 KiB
Bash
36 lines
1.4 KiB
Bash
TERMUX_PKG_HOMEPAGE=https://github.com/pytorch/vision
|
|
TERMUX_PKG_DESCRIPTION="Datasets, Transforms and Models specific to Computer Vision"
|
|
TERMUX_PKG_LICENSE="BSD 3-Clause"
|
|
TERMUX_PKG_MAINTAINER="@termux"
|
|
TERMUX_PKG_VERSION="0.25.0"
|
|
TERMUX_PKG_SRCURL="https://github.com/pytorch/vision/archive/refs/tags/v${TERMUX_PKG_VERSION}.tar.gz"
|
|
TERMUX_PKG_SHA256=a7ac1b3ab489d71f6e27edfad1e27616e4b8a9b1517e60fce4a950600d3510e8
|
|
TERMUX_PKG_DEPENDS="libc++, ffmpeg, python, python-numpy, python-pillow, python-pip, python-torch, libjpeg-turbo, libpng, libwebp, zlib"
|
|
TERMUX_PKG_SETUP_PYTHON=true
|
|
TERMUX_PKG_AUTO_UPDATE=true
|
|
|
|
termux_step_pre_configure() {
|
|
CXXFLAGS+=" -I${TERMUX_PYTHON_HOME}/site-packages/torch/include"
|
|
CXXFLAGS+=" -I${TERMUX_PYTHON_HOME}/site-packages/torch/include/torch/csrc/api/include"
|
|
CXXFLAGS+=" -DUSE_PYTHON"
|
|
LDFLAGS+=" -ltorch_cpu -ltorch_python -lc10"
|
|
|
|
# setting this $BUILD_PREFIX variable, which wasn't previously set, causes
|
|
# libwebp to be detected and libjpeg to be detected,
|
|
# and assists with detecting ffmpeg
|
|
export BUILD_PREFIX="$TERMUX_PREFIX"
|
|
export BUILD_VERSION=$TERMUX_PKG_VERSION
|
|
|
|
# this causes ffmpeg to be detected during cross-compilation,
|
|
# enabling the "video decoder extensions"
|
|
sed -i "s|shutil.which(\"ffmpeg\")|\"$TERMUX_PREFIX/bin/ffmpeg\"|" setup.py
|
|
}
|
|
|
|
termux_step_configure() {
|
|
:
|
|
}
|
|
|
|
termux_step_make_install() {
|
|
pip -v install --no-build-isolation --no-deps --prefix "$TERMUX_PREFIX" "$TERMUX_PKG_SRCDIR"
|
|
}
|