1
0
mirror of synced 2024-12-22 12:50:33 +01:00

draft ebuild for pytorch

This commit is contained in:
Horea Christian 2019-10-04 10:55:09 +02:00
parent 267dc793c3
commit 2798f806fb
2 changed files with 53 additions and 0 deletions

View File

@ -0,0 +1,17 @@
<?xml version='1.0' encoding='UTF-8'?>
<!DOCTYPE pkgmetadata SYSTEM "http://www.gentoo.org/dtd/metadata.dtd">
<pkgmetadata>
<maintainer type="person">
<email>chr@chymera.eu</email>
<name>Horea Christian</name>
</maintainer>
<maintainer type="project">
<email>sci@gentoo.org</email>
<name>Gentoo Science Project</name>
</maintainer>
<longdescription lang="en">
PyTorch is a Python package that provides two high-level features, namely
Tensor computation (like NumPy) with strong GPU acceleration and deep
neural networks built on a tape-based autograd system.
</longdescription>
</pkgmetadata>

View File

@ -0,0 +1,36 @@
# Copyright 1999-2018 Gentoo Authors
# Distributed under the terms of the GNU General Public License v2
EAPI=7
PYTHON_COMPAT=( python2_7 python3_6 )
inherit distutils-r1
DESCRIPTION="Tensors and Dynamic neural networks in Python with strong GPU acceleration"
HOMEPAGE="https://pytorch.org/"
SRC_URI="https://github.com/pytorch/${PN}/archive/v${PV}.tar.gz -> ${P}.tar.gz"
LICENSE="BSD"
SLOT="0"
KEYWORDS="~amd64 ~x86"
IUSE="test"
DEPEND="
test? ( dev-python/pytest[${PYTHON_USEDEP}] )
dev-python/setuptools[${PYTHON_USEDEP}]
"
RDEPEND="
dev-python/cffi[${PYTHON_USEDEP}]
dev-python/future[${PYTHON_USEDEP}]
dev-python/numpy[${PYTHON_USEDEP}]
dev-python/pyyaml[${PYTHON_USEDEP}]
dev-python/requests[${PYTHON_USEDEP}]
dev-python/six[${PYTHON_USEDEP}]
dev-python/typing[${PYTHON_USEDEP}]
dev-util/cmake
dev-util/ninja
sci-libs/mkl
|| ( dev-util/nvidia-cuda-toolkit dev-libs/cudnn )
"