diff options
Diffstat (limited to 'sci-libs/pytorch/pytorch-1.11.0.ebuild')
| -rw-r--r-- | sci-libs/pytorch/pytorch-1.11.0.ebuild | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/sci-libs/pytorch/pytorch-1.11.0.ebuild b/sci-libs/pytorch/pytorch-1.11.0.ebuild index 0a1cae78f4b..401bdea8264 100644 --- a/sci-libs/pytorch/pytorch-1.11.0.ebuild +++ b/sci-libs/pytorch/pytorch-1.11.0.ebuild @@ -16,11 +16,13 @@ LICENSE="BSD" SLOT="0" KEYWORDS="~amd64" RESTRICT="test" +IUSE="cuda ffmpeg nnpack +numpy opencl opencv openmp qnnpack xnnpack" REQUIRED_USE=${PYTHON_REQUIRED_USE} RDEPEND=" ${PYTHON_DEPS} ~sci-libs/caffe2-${PV}[${PYTHON_USEDEP}] + sci-libs/caffe2[cuda?,ffmpeg?,nnpack?,numpy?,opencl?,opencv?,openmp?,qnnpack?,xnnpack?] dev-python/typing-extensions[${PYTHON_USEDEP}] " DEPEND="${RDEPEND} |