diff options
| author |   Andrew Ammerlaan <andrewammerlaan@gentoo.org> | 2022-12-01 22:54:14 +0100 |
|---|---|---|
| committer | Andrew Ammerlaan <andrewammerlaan@gentoo.org> | 2022-12-01 22:54:36 +0100 |
| commit | bd5fc63025145b87d3627171a90d47aa241a76d5 (patch) | |
| tree | 1846daaddf21d00ab201e12fb5c4a35636dc52f8 /sci-libs/torchvision | |
| parent | dev-libs/oneCCL: new package, add 2021.7.1 (diff) | |
| download | sci-bd5fc63025145b87d3627171a90d47aa241a76d5.tar.gz sci-bd5fc63025145b87d3627171a90d47aa241a76d5.tar.bz2 sci-bd5fc63025145b87d3627171a90d47aa241a76d5.zip | |
sci-libs/torchvision: drop use cuda
it has been dropped on pytorch
Signed-off-by: Andrew Ammerlaan <andrewammerlaan@gentoo.org>
Diffstat (limited to 'sci-libs/torchvision')
| -rw-r--r-- | sci-libs/torchvision/metadata.xml | 3 | ||||
| -rw-r--r-- | sci-libs/torchvision/torchvision-0.11.2.ebuild | 10 |
2 files changed, 1 insertions, 12 deletions
diff --git a/sci-libs/torchvision/metadata.xml b/sci-libs/torchvision/metadata.xml index 28985da70..9c4ddaefd 100644 --- a/sci-libs/torchvision/metadata.xml +++ b/sci-libs/torchvision/metadata.xml @@ -5,9 +5,6 @@ <email>jpizarrocallejas@gmail.com</email> <name>Jorge Pizarro Callejas</name> </maintainer> - <use> - <flag name="cuda">Enable CUDA support if enabled in PyTorch</flag> - </use> <upstream> <remote-id type="github">pytorch/vision</remote-id> </upstream> diff --git a/sci-libs/torchvision/torchvision-0.11.2.ebuild b/sci-libs/torchvision/torchvision-0.11.2.ebuild index b17ac9007..765fd05b7 100644 --- a/sci-libs/torchvision/torchvision-0.11.2.ebuild +++ b/sci-libs/torchvision/torchvision-0.11.2.ebuild @@ -14,7 +14,6 @@ SRC_URI="https://github.com/pytorch/vision/archive/v${PV}.tar.gz -> ${P}.tar.gz" LICENSE="BSD" SLOT="0" KEYWORDS="~amd64" -IUSE="cuda" RDEPEND=" dev-python/av[${PYTHON_USEDEP}] @@ -23,7 +22,7 @@ RDEPEND=" dev-python/six[${PYTHON_USEDEP}] dev-python/tqdm[${PYTHON_USEDEP}] dev-python/scipy[${PYTHON_USEDEP}] - sci-libs/pytorch[cuda(-)?,${PYTHON_USEDEP}] + sci-libs/pytorch[${PYTHON_USEDEP}] media-video/ffmpeg dev-qt/qtcore:5 " @@ -38,10 +37,3 @@ BDEPEND=" S="${WORKDIR}/vision-${PV}" distutils_enable_tests pytest - -pkg_setup() { - if use cuda; then - export FORCE_CUDA=1 - export TORCH_CUDA_ARCH_LIST="3.5;3.7;5.0;5.2;5.3;6.0;6.0+PTX;6.1;6.1+PTX;6.2;6.2+PTX;7.0;7.0+PTX;7.2;7.2+PTX;7.5;7.5+PTX" - fi -} |