From mboxrd@z Thu Jan 1 00:00:00 1970 Return-Path: Received: from lists.gentoo.org (pigeon.gentoo.org [208.92.234.80]) (using TLSv1.3 with cipher TLS_AES_256_GCM_SHA384 (256/256 bits) key-exchange X25519 server-signature RSA-PSS (2048 bits)) (No client certificate requested) by finch.gentoo.org (Postfix) with ESMTPS id 7CF16158086 for ; Sun, 26 Dec 2021 14:26:38 +0000 (UTC) Received: from pigeon.gentoo.org (localhost [127.0.0.1]) by pigeon.gentoo.org (Postfix) with SMTP id C10C22BC025; Sun, 26 Dec 2021 14:26:37 +0000 (UTC) Received: from smtp.gentoo.org (woodpecker.gentoo.org [140.211.166.183]) (using TLSv1.3 with cipher TLS_AES_256_GCM_SHA384 (256/256 bits) key-exchange X25519 server-signature RSA-PSS (4096 bits) server-digest SHA256) (No client certificate requested) by pigeon.gentoo.org (Postfix) with ESMTPS id E19F42BC025 for ; Sun, 26 Dec 2021 14:26:35 +0000 (UTC) Received: from oystercatcher.gentoo.org (unknown [IPv6:2a01:4f8:202:4333:225:90ff:fed9:fc84]) (using TLSv1.3 with cipher TLS_AES_256_GCM_SHA384 (256/256 bits) key-exchange X25519 server-signature RSA-PSS (4096 bits)) (No client certificate requested) by smtp.gentoo.org (Postfix) with ESMTPS id C1911342CD9 for ; Sun, 26 Dec 2021 14:26:34 +0000 (UTC) Received: from localhost.localdomain (localhost [IPv6:::1]) by oystercatcher.gentoo.org (Postfix) with ESMTP id 72E601FB for ; Sun, 26 Dec 2021 14:26:33 +0000 (UTC) From: "Andrew Ammerlaan" To: gentoo-commits@lists.gentoo.org Content-Transfer-Encoding: 8bit Content-type: text/plain; charset=UTF-8 Reply-To: gentoo-dev@lists.gentoo.org, "Andrew Ammerlaan" Message-ID: <1640528785.039d53dff3102d3ac7bdcc3d102c68459a084439.andrewammerlaan@gentoo> Subject: [gentoo-commits] proj/sci:master commit in: sci-libs/torchvision/ X-VCS-Repository: proj/sci X-VCS-Files: sci-libs/torchvision/metadata.xml sci-libs/torchvision/torchvision-0.11.2.ebuild X-VCS-Directories: sci-libs/torchvision/ X-VCS-Committer: andrewammerlaan X-VCS-Committer-Name: Andrew Ammerlaan X-VCS-Revision: 039d53dff3102d3ac7bdcc3d102c68459a084439 X-VCS-Branch: master Date: Sun, 26 Dec 2021 14:26:33 +0000 (UTC) Precedence: bulk List-Post: List-Help: List-Unsubscribe: List-Subscribe: List-Id: Gentoo Linux mail X-BeenThere: gentoo-commits@lists.gentoo.org X-Auto-Response-Suppress: DR, RN, NRN, OOF, AutoReply X-Archives-Salt: 4f3e9f93-38ee-486b-a8b2-83c6002dd21d X-Archives-Hash: edc4ad41a3c72c91313162bc5190c086 commit: 039d53dff3102d3ac7bdcc3d102c68459a084439 Author: Andrew Ammerlaan gentoo org> AuthorDate: Sun Dec 26 14:26:25 2021 +0000 Commit: Andrew Ammerlaan gentoo org> CommitDate: Sun Dec 26 14:26:25 2021 +0000 URL: https://gitweb.gentoo.org/proj/sci.git/commit/?id=039d53df sci-libs/torchvision: new package Closes: https://github.com/gentoo/sci/issues/943 Closes: https://github.com/gentoo/sci/pull/944 Package-Manager: Portage-3.0.30, Repoman-3.0.3 Signed-off-by: Andrew Ammerlaan gentoo.org> sci-libs/torchvision/metadata.xml | 11 +++++++ sci-libs/torchvision/torchvision-0.11.2.ebuild | 45 ++++++++++++++++++++++++++ 2 files changed, 56 insertions(+) diff --git a/sci-libs/torchvision/metadata.xml b/sci-libs/torchvision/metadata.xml new file mode 100644 index 000000000..ad3ef354b --- /dev/null +++ b/sci-libs/torchvision/metadata.xml @@ -0,0 +1,11 @@ + + + + + jpizarrocallejas@gmail.com + Jorge Pizarro Callejas + + + Enable CUDA support if enabled in PyTorch + + diff --git a/sci-libs/torchvision/torchvision-0.11.2.ebuild b/sci-libs/torchvision/torchvision-0.11.2.ebuild new file mode 100644 index 000000000..6c58eeb6e --- /dev/null +++ b/sci-libs/torchvision/torchvision-0.11.2.ebuild @@ -0,0 +1,45 @@ +# Copyright 2020-2021 Gentoo Authors +# Distributed under the terms of the GNU General Public License v2 + +EAPI=8 + +PYTHON_COMPAT=( python3_10 ) + +inherit distutils-r1 + +DESCRIPTION="Datasets, transforms and models to specific to computer vision" +HOMEPAGE="https://github.com/pytorch/vision" +SRC_URI="https://github.com/pytorch/vision/archive/v${PV}.tar.gz -> ${P}.tar.gz" + +LICENSE="BSD" +SLOT="0" +KEYWORDS="~amd64" +IUSE="cuda test" + +RDEPEND=" + dev-python/av[${PYTHON_USEDEP}] + dev-python/numpy[${PYTHON_USEDEP}] + dev-python/pillow[${PYTHON_USEDEP}] + dev-python/six[${PYTHON_USEDEP}] + dev-python/tqdm[${PYTHON_USEDEP}] + dev-python/scipy[${PYTHON_USEDEP}] + sci-libs/pytorch[cuda?,python,${PYTHON_USEDEP}] + media-video/ffmpeg + dev-qt/qtcore:5 +" +DEPEND="${RDEPEND}" +BDEPEND=" + test? ( + dev-python/mock[${PYTHON_USEDEP}] + )" + +S="${WORKDIR}/vision-${PV}" + +distutils_enable_tests pytest + +pkg_setup() { + if use cuda; then + export FORCE_CUDA=1 + export TORCH_CUDA_ARCH_LIST="3.5;3.7;5.0;5.2;5.3;6.0;6.0+PTX;6.1;6.1+PTX;6.2;6.2+PTX;7.0;7.0+PTX;7.2;7.2+PTX;7.5;7.5+PTX" + fi +}