March 24, 2021 - With the PyTorch 1.8 release, we are delighted to announce a new installation option for users of PyTorch on the ROCm™ open software platform. An installable Python package is now hosted on pytorch.org, along with instructions for local installation in the same simple, selectable format as PyTorch packages for CPU-only configurations and other GPU platforms.

Details for the file cupy_rocm_5_0-13.3.0-cp310-cp310-manylinux2014_x86_64.whl. Download URL: cupy_rocm_5_0-13.3.0-cp310-cp310-manylinux2014_x86_64.whl

python package of rocm-smi-lib. Contribute to ROCm/pyrsmi development by creating an account on GitHub.

Details for the file cupy_rocm_4_0-10.6.0-cp38-cp38-manylinux1_x86_64.whl. Download URL: cupy_rocm_4_0-10.6.0-cp38-cp38-manylinux1_x86_64.whl

To access this feature, go to pytorch.org/get-started/locally/. For the correct wheels command, you must select Linux, Python, pip, and ROCm in the matrix. ... The available ROCm release varies between the PyTorch Build of Stable or Nightly. More recent releases are generally available through ...

Details for the file tensorflow_rocm-2.14.0.600-cp39-cp39-manylinux2014_x86_64.whl. Download URL: tensorflow_rocm-2.14.0.600-cp39-cp39-manylinux2014_x86_64.whl

$ pip install torch==2.2.0.dev20231206+rocm5.7 --index-url https://download.pytorch.org/whl/nightly/rocm5.7 # tested version ... Install ROCm’s flash attention (v2.0.4) following the instructions from ROCmSoftwarePlatform/flash-attention ... If you are using rocm5.7 with pytorch 2.1.0 onwards, ...

The ROCm HIP SDK including the ROCm LLVM Compiler's C-API headers. Python 3.8+.

pip install --no-cache-dir ` https://repo.radeon.com/rocm/windows/rocm-rel-7.2.1/torch-2.9.1+rocm7.2.1-cp312-cp312-win_amd64.whl ` https://repo.radeon.com/rocm/windows/rocm-rel-7.2.1/torchaudio-2.9.1+rocm7.2.1-cp312-cp312-win_amd64.whl ` https://repo.radeon.com/rocm/windows/rocm-rel-7.2.1/torchvision-0.24.1+rocm7.2.1-cp312-cp312-win_amd64.whl · Confirm if PyTorch is correctly installed. Verify if Pytorch is installed and detecting the GPU compute device. python -c "import torch" 2>nul && echo Success || echo Failure

November 21, 2025 - If you find ready rocm_sdk and pytorch builds for your architecture and OS, lucky you—download the .whl packages and jump to "Installing ComfyUI with ROCm Support".

For details see # https://github.com/ROCm/TheRock/blob/main/docs/environment_setup_guide.md#patchelf sudo apt install curl make sudo env INSTALL_PREFIX=/usr/local ./dockerfiles/install_pinned_patchelf.sh # Init python virtual environment and install python dependencies python3 -m venv .venv && source .venv/bin/activate pip install --upgrade pip pip install -r requirements.txt # Download submodules and apply patches python3 ./build_tools/fetch_sources.py ·

Install ROCM. Install pip, virtual environment and development headers for Python 3:

sudo zypper in rocm-runtime \ miopen rccl rocblas amdsmi \ hipblaslt hiprand hipcub \ hipsolver hipfft rocm-cmake \ rocm-compilersupport \ rocm-llvm-filesystem \ rocm-clang-runtime-devel \ hipcub-devel rocm-hip-devel \ libhipfft0-devel libhipsolver0-devel \ libhipsparse1-devel rocthrust-devel \ librocfft0 rocm-core rocrand rocsolver · This procedure should also work for Leap-based distributions and Slowroll (adjust the distro-specific lines), but it is not tested. Note: This also installs build dependencies for flash-attention. ... sudo pacman -S base-devel python-pip python-virtualenv git clone https://aur.archlinux.org/python312.git cd python312 makepkg -si cd ..

March 13, 2024 - Then you can check if ROCm was ... running rocminfo. If an error is returned then something went wrong with the installation. Another possibility is that secure boot may cause issues on some systems, so if you received an error here then disabling secure boot may help. Next we'll download PyTorch with PIP in a Python virtual ...

The ROCm WHLs available at PyTorch.org are not tested extensively by AMD as the WHLs change regularly when the nightly builds are updated. Important! When manually downloading WHLs from repo.radeon, ensure to select the compatible WHLs for specific Python versions.

echo 'deb [arch=amd64] https://repo.radeon.com/rocm/apt/<release> ubuntu main' | sudo tee /etc/apt/sources.list.d/rocm.list sudo apt update

ROCm Docs Core is also distributed as a pip package available from PyPi as rocm-docs-core

You can install CuPy package via pip. It builds CuPy from source. $ export HCC_AMDGPU_TARGET=gfx900 # This value should be changed based on your GPU $ export __HIP_PLATFORM_HCC__ $ export CUPY_INSTALL_USE_HIP=1 $ pip install cupy · The tarball of the source tree is available via pip download ...

Learn about using the ROCm SMI library with Python.