Provided ASIC support, users can now set a maximum or minimum sclk or mclk value through our Python CLI tool (rocm-smi –setextremum max sclk 1500).

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.

This repository provides low-level Python and Cython Bindings for HIP and an interoperability layer for CUDA® Python programs (Python and Cython). Currently, only Linux is supported (prebuilt packages and code). Prebuilt packages distributed via PyPI (or Test PyPI) are only provided for Linux systems that agree with the manylinux_2_17_x86_64 tag. Requires that a compatible ROCm™ HIP SDK is installed on your system.

pyrsmi is a Python package providing system management functionality for AMD GPUs. Version 1.0+ uses the new AMD SMI Library (amdsmi), replacing the deprecated rocm-smi-lib.

This repository provides low-level Python and Cython Bindings for ROCm LLVM.

November 21, 2025 - In this article, I’ll share my experience building the yet-to-be-released (at the time of writing) ROCm 7 for my own GPUs, even if they’re not in the list of officially supported architectures. Then, I’ll show how to build PyTorch with the assembled ROCm 7 and run ComfyUI.

For simple workloads and/or prototyping AMD creates a Docker Images based on Ubuntu using the latest ROCm release and Supported ROCm-Pytorch builds found at ROCM Dockerhub. The intent is to get users up and running with their custom workload in python and provides an environment of prebuild ROCm, Onnxruntime and MIGraphX packages required to get started without the need to build Onnxruntime.

1 month ago - ROCm is also enabling application development on some of the world’s top supercomputers. ... Supports multiple programming language choices for HPC workloads, giving developers a range of tools to help solve the problem at hand—including OpenMP®, HIP, OpenCL™, and Python™

Listing 14 Compiling a Cython module that uses HIP Python’s Cython API.# 1import os, sys 2 3mymodule = "mymodule" 4 5from setuptools import Extension, setup 6from Cython.Build import cythonize 7 8ROCM_PATH=os.environ.get("ROCM_PATH", "/opt/rocm") 9HIP_PLATFORM = os.environ.get("HIP_PLATFORM", "amd") 10 11if HIP_PLATFORM not in ("amd", "hcc"): 12 raise RuntimeError("Currently only HIP_PLATFORM=amd is supported") 13 14def create_extension(name, sources): 15 global ROCM_PATH 16 global HIP_PLATFORM 17 rocm_inc = os.path.join(ROCM_PATH,"include") 18 rocm_lib_dir = os.path.join(ROCM_PATH,"lib") 19

import sys sys.path.append("/opt/rocm/libexec/rocm_smi/") try: import rocm_smi except ImportError: raise ImportError("Could not import /opt/rocm/libexec/rocm_smi/rocm_smi.py") class prof_utils: def __init__(self, mode) -> None: rocm_smi.initializeRsmi() def getPower(self, device): return rocm_smi.getPower(device) def listDevices(self): return rocm_smi.listDevices() def getMemInfo(self, device): (memUsed, memTotal) = rocm_smi.getMemInfo(device, "vram") return round(float(memUsed)/float(memTotal) * 100, 2)

Provided ASIC support, users can now set a maximum or minimum sclk or mclk value through our Python CLI tool (rocm-smi –setextremum max sclk 1500).

You can install the rocm Python package for any architecture inside a venv and run offload-arch from there:

CuPy is a NumPy/SciPy-compatible array library for GPU-accelerated computing with Python. This is a CuPy wheel (precompiled binary) package for AMD ROCm 5.0.

pip install tensorflow-rocm Copy PIP instructions · Latest release · Released: Jan 10, 2024 · TensorFlow is an open source machine learning framework for everyone. These details have been verified by PyPI · JasonF-amd rsanthanam-amd sunway513 · These details have not been verified by PyPI · Homepage · Download · License: Apache Software License (Apache 2.0) Author: Google Inc. Tags tensorflow , tensor , machine , learning · Requires: Python >=3.9 ·

cd .ci/docker ./build.sh pytorch-linux-<os-version>-rocm<rocm-version>-py<python-version> -t rocm/pytorch:build_from_dockerfile

ROCm Documentation Python package for ReadTheDocs build standardization - ROCm/rocm-docs-core

--setextremum <min/max> <sclk or mclk> <value in MHz to set to> Provided ASIC support, users can now set a maximum or minimum sclk or mclk value through our Python CLI tool (rocm-smi --setextremum max sclk 1500).

June 23, 2023 - HIP Python provides low-level Cython and Python® bindings for the HIP runtime, HIPRTC, multiple math libraries and the communication library RCCL, and further a CUDA® Python Interoperability layer that aims to simplify the porting of CUDA Python Python and Cython programs. ... This documentation has been generated based on ROCm™ version 5.7.1.

This section describes the ROCm SMI Python module API.