Installation Guide#

Advanced Installation#

The rest of this guide gives more specific installation instructions for:

  • Set-up on other operating systems, such as a Microsoft Windows install, using Microsoft Visual Studio.

  • Installation of PyTorch with GPU back-ends for better performance.

  • Installing Quark with usage examples.

  • Pre-compiling kernels and operators.

  • Validating the installation.

Install Python#

Python 3.9, 3.10, 3.11 or 3.12 is required. Python 3.13 is not currently supported by Quark’s dependencies.

On all platforms we recommend installing Python with an environment such as Miniconda, which will simplify installation of dependencies.

Note

On Windows consider adding Conda to your PATH in the install options, which will let you activate your environments easily, from within the Developer Command Prompt.

If you are using a Python environment, make sure that you create and activate an environment for Quark, and that you install the dependencies from the following subsections into that environment. e.g.

conda create -n AMD_Quark python=3.12
conda activate AMD_Quark

Verify your Python version is one of those supported with:

python --version

You should see the version number returned e.g. Python 3.12.9, which is fine.

If you are using an Anaconda-based environment, such as Miniconda, we recommend that you install the following dependencies with pip install.

Install PyTorch with GPU Support#

PyTorch 2.2.0 or later is required.

Windows#

To install PyTorch with CUDA 12.6 GPU support, in a Python environment using pip:

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu126

If CUDA is not available, install PyTorch without GPU support:

pip install torch torchvision torchaudio

Linux#

To install PyTorch with ROCm 6.2.4 GPU support, in a Python environment using pip:

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.2.4

To install PyTorch with CUDA 12.6 GPU support:

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu126

If neither of these combinations is available on your system, you may install without GPU support:

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu

For further options, see the PyTorch install guide.

Install a C++ Compiler#

Windows#

When installing on Windows, Visual Studio is required, with Visual Studio 2022 being the minimum required version. During the compilation process, you can either use the Developer Command Prompt, or manually add paths to environment variables.

When installing Visual Studio, ensure that you include the Developer Command Prompt in the installation. You can access it through Visual Studio by selecting Tools > Command Line > Developer Command Line, or as a profile tab in the Windows Terminal application, installed through the Microsoft Store.

Alternatively, if you do not wish to use the Developer Command Prompt, you can also manually add tool paths to PATH environment variable. The tools cl.exe, MSBuild.exe, and link.exe from Visual Studio are used. These programs can be found in the Visual Studio installation directory. In the Edit Environment Variables window, click New, and then paste the path to the folder containing the cl.exe, link.exe, and MSBuild.exe files. Click OK on all of the windows to apply the changes.

Linux#

On Ubuntu the g++ compiler is installed with the build-essential package. Note that while many versions of C++ compiler may work, we currently only confirm support for g++ version 13.3, which installs with the build-essential package on Ubuntu 24.04.

Install CMake#

You will need CMake installed on your PATH. One option here is to install it into your Python environment with:

pip install cmake

Install Quark#

Note

The AMD Quark package distribution name has been renamed to amd-quark. Please use the new package name for releases newer than 0.6.0.

We recommend new users install Quark from PyPI with pip. It’s also possible to install from a ZIP download, which contains additional examples.

Install Quark from PyPI with pip#

Releases of AMD Quark are available on PyPI at https://pypi.org/project/amd-quark/, and can be installed with pip:

pip install amd-quark

Nightly builds are not yet available on PyPI.

Install Quark + Quark Examples from Download#

Download and unzip 📥*amd_quark-*.zip*, which has a wheel package in it. You can also download the wheel package 📥*amd_quark-*.whl* directly. We strongly recommend downloading the ZIP file, as it includes examples compatible with the wheel package version.

Directory Structure of the zip file:

+ amd_quark.zip
   + amd_quark.whl
   + examples    # Examples of using Quark.
   + docs        # Off-line documentation of Quark.
   + README.md

Then install the quark wheel package by running the following command:

pip install amd_quark*.whl

Note

If your Quark pip install fails with dependency version mismatches, check that you are running a supported version of Python.

Note

If your Windows pip installation of ONNX dependencies with Quark is failing on a long generated path name, you may enable long path name support in Windows in the Group Policy Editor. In the Windows Start Menu, type GPEDIT.MSC in search box, and open the editor. Navigate to: Computer Configuration -> Administrative Templates -> System -> Filesystem, and change the setting for Enable Win32 long paths to Enabled. You will need to restart your terminal for changes to take effect.

Installation Verification (Optional)#

Verify the installation by running:

python -c "import quark"

If no error is reported, then the installation was successful.

Compile Fast Quantization Kernels (Optional)#

When using Quark’s quantization APIs for the first time, it compiles the fast quantization kernels using your installed PyTorch with GPU support, if available. This process might take a few minutes, but the subsequent quantization calls are then much faster. This process requires the Transformers package from Hugging Face. To invoke this compilation now, and check if it is successful, run the following commands:

pip install transformers
python -c "import quark.torch.kernel"

If the kernel cannot be loaded successfully with Quark on Windows with GPU support, follow the steps below to troubleshoot:

  • Ensure a C++ compiler is installed, and can be invoked from the command line.

  • Check GPU support with a Python call to torch.cuda.is_available() returning True.

  • Verify that nvcc, for CUDA, or hipcc, for ROCm HIP, can be invoked from the command line.

  • If the compilation is successful, but Python fails to load the DLL, locate the Quark build directory (the build path will be printed in the log), and check the dependencies using dumpbin /DEPENDENTS kernel_ext.pyd.

  • Check that the required DLL is included in the system path.

  • Check that the version of the dependent DLL is correct.

  • Check that the Python version of the dependent DLL is correct.

Compile Custom Operators Library (Optional)#

When using Quark’s ONNX custom operators for the first time, it compiles the custom operators library using your local environment. To invoke this compilation now, and check if it is successful, run the following command:

python -c "import quark.onnx.operators.custom_ops"

Previous Versions of AMD Quark#

Note: The following links are for older versions of AMD Quark, before the package distribution name was renamed to amd-quark.