Skip to content

AMD GPU Support

Install AMD GPU version Torch

pip3 install torch torchvision --index-url https://download.pytorch.org/whl/rocm7.1 
pip3 install --pre torchao --index-url https://download.pytorch.org/whl/rocm7.1

Then, you can install cache-dit and other libraries as usual, please refer to the Installation section and Examples in the user guide for more details.