Arch Linux User Repository

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

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.

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

@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

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.

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