2019-10-09 18:26:39 +02:00
|
|
|
# Copyright 1999-2019 Gentoo Authors
|
2019-10-04 10:55:09 +02:00
|
|
|
# Distributed under the terms of the GNU General Public License v2
|
|
|
|
|
|
|
|
|
|
EAPI=7
|
|
|
|
|
|
|
|
|
|
PYTHON_COMPAT=( python2_7 python3_6 )
|
|
|
|
|
|
2019-10-09 18:26:39 +02:00
|
|
|
inherit cmake-utils distutils-r1
|
2019-10-04 10:55:09 +02:00
|
|
|
|
|
|
|
|
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"
|
2019-10-09 18:26:39 +02:00
|
|
|
KEYWORDS="~amd64"
|
2019-10-04 10:55:09 +02:00
|
|
|
IUSE="test"
|
|
|
|
|
|
|
|
|
|
DEPEND="
|
|
|
|
|
test? ( dev-python/pytest[${PYTHON_USEDEP}] )
|
|
|
|
|
dev-python/setuptools[${PYTHON_USEDEP}]
|
2019-10-09 18:26:39 +02:00
|
|
|
"
|
2019-10-04 10:55:09 +02:00
|
|
|
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/ninja
|
2019-10-09 18:26:39 +02:00
|
|
|
sci-libs/mkl-rt
|
2019-10-04 10:55:09 +02:00
|
|
|
|| ( dev-util/nvidia-cuda-toolkit dev-libs/cudnn )
|
2019-10-09 18:26:39 +02:00
|
|
|
"
|
