commit:     e23377f782e7184f960a704d4bbe48c514f7831c
Author:     Andrew Ammerlaan <andrewammerlaan <AT> gentoo <DOT> org>
AuthorDate: Tue Jul  5 14:40:32 2022 +0000
Commit:     Andrew Ammerlaan <andrewammerlaan <AT> gentoo <DOT> org>
CommitDate: Tue Jul  5 14:40:32 2022 +0000
URL:        https://gitweb.gentoo.org/proj/sci.git/commit/?id=e23377f7

sci-libs/torchvision: depend on pytorch[cuda?]

Signed-off-by: Andrew Ammerlaan <andrewammerlaan <AT> gentoo.org>

 sci-libs/torchvision/torchvision-0.11.2.ebuild | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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
 "

Reply via email to