summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
Diffstat (limited to 'sci-libs/pytorch/pytorch-1.11.0.ebuild')
-rw-r--r--sci-libs/pytorch/pytorch-1.11.0.ebuild2
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}