Skip to content

[Bug] amd rocm7.14, but the installer opted to download 7.2 #7264

Description

@bat3a

Note: Please do not remove the questions. Answer beside them.

  1. Did you update? pip install --upgrade unsloth unsloth_zoo first time install
  2. Colab or Kaggle or local / cloud :local
  3. Number GPUs used, use nvidia-smi strix halo box
  4. Which notebook? Please link! none
  5. Which Unsloth version, TRL version, transformers version, PyTorch version?
  6. Which trainer? SFTTrainer, GRPOTrainer etc none
Put Minimal code to reproduce error here ###Remove Hugging Face token###
###Please make sure to check formatting properly, edit if needed.###

just first time install, although it found the amd rocm7.14, but the installer opted to download 7.2!

me@meMax:~$ curl -fsSL https://unsloth.ai/install.sh | sh

  🦥 Unsloth Studio Installer
  ────────────────────────────────────────────────────

  platform       linux
  deps           all system dependencies found
                 preserving existing environment for rollback...
                 previous environment preserved for rollback
  venv           creating Python 3.13 virtual environment
                 /home/me/.unsloth/studio/unsloth_studio
  venv           using environment
                 /home/me/.unsloth/studio/unsloth_studio
                 existing install has torch 2.11.0+rocm7.2 -- keeping it (set UNSLOTH_TORCH_UPGRADE=1 to get the newest release)
  gpu            AMD ROCm (gfx1151)
                 ROCm: /opt/rocm
                 hipconfig: 7.14.60850-0000000
                 GPU: AMD RYZEN AI MAX+ 395 w/ Radeon 8060S
                 wheels: repo.radeon.com (Radeon)
                 [WARN] Radeon repo unavailable; falling back to ROCm index (https://download.pytorch.org/whl/rocm7.2)
                 installing bitsandbytes for AMD ROCm (pre-release, PR #1887)...
                 installing unsloth (this may take a few minutes)...

>>> print("PyTorch Version:", torch.__version__)
PyTorch Version: 2.11.0+rocm7.14.0


🦥 You can also ask via our Reddit page: https://reddit.com/r/unsloth/

Metadata

Metadata

Assignees

No one assigned

    Labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions