diff --git a/dev-python/pytorch/metadata.xml b/dev-python/pytorch/metadata.xml new file mode 100644 index 0000000..3134318 --- /dev/null +++ b/dev-python/pytorch/metadata.xml @@ -0,0 +1,17 @@ + + + + + chr@chymera.eu + Horea Christian + + + sci@gentoo.org + Gentoo Science Project + + + 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. + + diff --git a/dev-python/pytorch/pytorch-1.2.0.ebuild b/dev-python/pytorch/pytorch-1.2.0.ebuild new file mode 100644 index 0000000..35343e7 --- /dev/null +++ b/dev-python/pytorch/pytorch-1.2.0.ebuild @@ -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 ) + " +