requirements-rocm.txt

# For AMD GPU (ROCm) support.
# Install with:
#   pip install -r requirements-rocm.txt
#   pip uninstall onnxruntime -y && pip install onnxruntime-rocm
#
# Requires ROCm drivers: https://rocm.docs.amd.com/en/latest/deploy/linux/index.html
# Supported GPUs: RX 6000/7000/9000 series, Instinct MI series

--extra-index-url https://download.pytorch.org/whl/rocm6.4
-r requirements.txt

# ROCm builds of PyTorch (override CPU/CUDA versions from requirements.txt)
torch
torchvision

# Note: onnxruntime from requirements.txt does not support ROCm.
# After installing, replace it manually:
#   pip uninstall onnxruntime -y && pip install onnxruntime-rocm

# Note: ctranslate2 does not support ROCm.
# Use m2m100_hf / nllb instead of m2m100 / nllb for offline translation on ROCm.