aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
Diffstat (limited to 'sci-libs/torchvision/torchvision-0.11.2.ebuild')
-rw-r--r--sci-libs/torchvision/torchvision-0.11.2.ebuild2
1 files changed, 1 insertions, 1 deletions
diff --git a/sci-libs/torchvision/torchvision-0.11.2.ebuild b/sci-libs/torchvision/torchvision-0.11.2.ebuild
index 6df2cbc01..b17ac9007 100644
--- a/sci-libs/torchvision/torchvision-0.11.2.ebuild
+++ b/sci-libs/torchvision/torchvision-0.11.2.ebuild
@@ -23,7 +23,7 @@ RDEPEND="
dev-python/six[${PYTHON_USEDEP}]
dev-python/tqdm[${PYTHON_USEDEP}]
dev-python/scipy[${PYTHON_USEDEP}]
- sci-libs/pytorch[${PYTHON_USEDEP}]
+ sci-libs/pytorch[cuda(-)?,${PYTHON_USEDEP}]
media-video/ffmpeg
dev-qt/qtcore:5
"