Package Details: python-torchvision-rocm 0.28.0-1

Git Clone URL: https://aur.archlinux.org/python-torchvision-rocm.git (read-only, click to copy)
Package Base: python-torchvision-rocm
Description: Datasets, transforms, and models specific to computer vision (with ROCM support)
Upstream URL: https://github.com/pytorch/vision
Licenses: BSD-3-Clause
Conflicts: python-torchvision
Provides: python-torchvision
Submitter: wuxxin
Maintainer: wuxxin
Last Packager: wuxxin
Votes: 12
Popularity: 0.097240
First Submitted: 2022-04-18 13:21 (UTC)
Last Updated: 2026-07-14 15:30 (UTC)

Required by (102)

Sources (2)

Pinned Comments

wuxxin commented on 2022-04-18 13:29 (UTC) (edited on 2026-02-04 14:51 (UTC) by wuxxin)

Please open issues and PRs at https://github.com/wuxxin/aur-packages

Latest Comments

« First ‹ Previous 1 2

0000000011111111 commented on 2023-05-19 14:32 (UTC) (edited on 2023-05-19 14:32 (UTC) by 0000000011111111)

It fails on AMD because there are 2 lines with FORCE_CUDA=1 in the PKGBUILD, which makes it only work on NVIDIA.

banditsan commented on 2023-04-18 22:18 (UTC)

@TheJackiMonster are you trying use package with nvidia GPU? This package is for AMD ROCm. For cuda there is torchvision-cuda in official Arch repo

TheJackiMonster commented on 2023-04-10 22:53 (UTC)

CUDA seems to be required to build the package: https://archlinux.org/packages/community/x86_64/cuda/

Also: "CUDA_HOME environment variable is not set. Please set it to your CUDA install root."

I would recommend adding cuda to the build requirements and set CUDA_HOME appropiately.

wuxxin commented on 2022-04-18 13:29 (UTC) (edited on 2026-02-04 14:51 (UTC) by wuxxin)

Please open issues and PRs at https://github.com/wuxxin/aur-packages