+1 for dreieck's request, and if you could please also add gfx90c
we'd be grateful!
Search Criteria
Package Details: python-torchvision-rocm 0.17.2-1
Package Actions
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: | 4 |
Popularity: | 0.63 |
First Submitted: | 2022-04-18 13:21 (UTC) |
Last Updated: | 2024-04-24 20:53 (UTC) |
Dependencies (16)
- ffmpeg (ffmpeg-intel-full-gitAUR, ffmpeg-nvcodec-11-1-gitAUR, ffmpeg-gitAUR, ffmpeg-cudaAUR, ffmpeg-decklinkAUR, ffmpeg-amd-fullAUR, ffmpeg-full-gitAUR, ffmpeg-nocudaAUR, ffmpeg-ffplayoutAUR, ffmpeg-obsAUR, ffmpeg-amd-full-gitAUR, ffmpeg-fullAUR, ffmpeg-headlessAUR, ffmpeg-libfdk_aacAUR)
- libjpeg-turbo (mozjpeg-gitAUR, libjpeg-turbo-gitAUR, mozjpegAUR)
- libpng (libpng-gitAUR, libpng-apngAUR)
- numactl (numactl-gitAUR)
- python-numpy (python-numpy-flameAUR, python-numpy-mkl-binAUR, python-numpy-gitAUR, python-numpy-mklAUR)
- python-pillow (python-pillow-gitAUR)
- python-pytorch-rocm (python-pytorch-cxx11abi-opt-rocmAUR, python-pytorch-opt-rocm)
- python-requests
- python-scipy (python-scipy-mkl-binAUR, python-scipy-gitAUR, python-scipy-mklAUR)
- python-sympy (python-sympy-gitAUR)
- cmake (cmake-gitAUR) (make)
- miopen-hip (opencl-amd-devAUR) (make)
- ninja (ninja-kitwareAUR, ninja-memAUR, ninja-fuchsia-gitAUR, ninja-gitAUR) (make)
- python-setuptools (make)
- qt6-base (qt6-base-gitAUR, qt6-base-headlessAUR) (make)
- rocm-hip-sdk (opencl-amd-devAUR) (make)
Required by (63)
- ballontranslator-git (requires python-torchvision) (optional)
- chainner-bin (requires python-torchvision) (optional)
- monailabel (requires python-torchvision)
- panelcleaner (requires python-torchvision)
- python-basicsr (requires python-torchvision)
- python-basicsr-cuda (requires python-torchvision)
- python-bitsandbytes-cuda121 (requires python-torchvision)
- python-clean-fid-git (requires python-torchvision)
- python-compressai (requires python-torchvision)
- python-compressai-vision (requires python-torchvision)
- python-d2l (requires python-torchvision) (optional)
- python-easyocr (requires python-torchvision)
- python-easyocr (requires python-torchvision) (check)
- python-easyocr-git (requires python-torchvision)
- python-effdet (requires python-torchvision)
- python-facexlib (requires python-torchvision)
- python-fairscale (requires python-torchvision)
- python-fastai (requires python-torchvision)
- python-fastai2 (requires python-torchvision)
- python-garage (requires python-torchvision) (optional)
- Show 43 more...
Sources (4)
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.