Package Details: python-torchvision-rocm 0.16.2-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
Conflicts: python-pytorch, python-torchvision
Provides: python-torchvision
Submitter: wuxxin
Maintainer: wuxxin (acxz)
Last Packager: wuxxin
Votes: 4
Popularity: 0.42
First Submitted: 2022-04-18 13:21 (UTC)
Last Updated: 2023-12-20 15:33 (UTC)

Required by (62)

Sources (2)

Pinned Comments

wuxxin commented on 2022-04-18 13:29 (UTC)

Development is on Github: https://github.com/rocm-arch/python-torchvision-rocm Please open issues and PRs there instead of commenting.

Latest Comments

Felix_F commented on 2023-10-17 16:38 (UTC)

Just in case anyone encounters the same issue as me, I'll leave my solution here:

I was trying to install python-torchvision-rocm, however since I had opencl-amd installed, that provided the librocm_smi64.so binary. The version of this one is at librocm_smi64.so.5, while python-torchvision-rocm expects librocm_smi64.so.1, so it was throwing this error: ImportError: librocm_smi64.so.1: cannot open shared object file: No such file or directory.

To fix this, I simply added a symlink from librocm_smi64.so.1 to the currently installed binary (in my case librocm_smi64.so.5.0.50700 located in /opt/rocm/lib/) by running sudo ln -s librocm_smi64.so.5.0.50700 /opt/rocm/lib/librocm_smi64.so.1.

Probably not the cleanest approach and an update might break it, however it works for now. If anyone knows about a better way to fix this, feel free to reply.

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)

Development is on Github: https://github.com/rocm-arch/python-torchvision-rocm Please open issues and PRs there instead of commenting.