+1 for dreieck's request, and if you could please also add gfx90c
we'd be grateful!
Search Criteria
Package Base Details: python-torchvision-rocm
Git Clone URL: | https://aur.archlinux.org/python-torchvision-rocm.git (read-only, click to copy) |
---|---|
Submitter: | wuxxin |
Maintainer: | wuxxin |
Last Packager: | wuxxin |
Votes: | 4 |
Popularity: | 0.51 |
First Submitted: | 2022-04-18 13:21 (UTC) |
Last Updated: | 2024-04-24 20:53 (UTC) |
Packages (2)
Latest Comments
fastrespawn commented on 2024-04-02 01:49 (UTC)
dreieck commented on 2024-03-26 11:57 (UTC)
Can you please add further AMD GPUs to _PYTORCH_ROCM_ARCH
?
Mine e.g. is gfx1103
. See here for a list.
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) (edited on 2024-03-20 18:27 (UTC) by wuxxin)
Development is on Github: https://github.com/wuxxin/aur-packages Please open issues and PRs there instead of commenting.
Pinned Comments
wuxxin commented on 2022-04-18 13:29 (UTC) (edited on 2024-03-20 18:27 (UTC) by wuxxin)
Development is on Github: https://github.com/wuxxin/aur-packages Please open issues and PRs there instead of commenting.