37 lines
902 B
Bash
37 lines
902 B
Bash
|
# 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 )
|
||
|
"
|
||
|
|