aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorAndrew Ammerlaan <andrewammerlaan@gentoo.org>2022-07-05 16:40:32 +0200
committerAndrew Ammerlaan <andrewammerlaan@gentoo.org>2022-07-05 16:40:32 +0200
commite23377f782e7184f960a704d4bbe48c514f7831c (patch)
tree1e0119127045cb97748cf3c229c45a3a1a7296aa
parentdev-python/zarr: skipping tests reported upstream (diff)
downloadsci-e23377f782e7184f960a704d4bbe48c514f7831c.tar.gz
sci-e23377f782e7184f960a704d4bbe48c514f7831c.tar.bz2
sci-e23377f782e7184f960a704d4bbe48c514f7831c.zip
sci-libs/torchvision: depend on pytorch[cuda?]
Signed-off-by: Andrew Ammerlaan <andrewammerlaan@gentoo.org>
-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
"