Skip to content

Ampere-specific pip install line gives error message (I have a 4090 RTX) #35

@araleza

Description

@araleza

Hello, I tried to follow the pip installation instructions for unslothai. I have a 4090 RTX, so I used the ampere command:

pip install "unsloth[cu118_ampere] @ git+https://github.com/unslothai/unsloth.git"

This produced the error messages shown below. I also found that if I used the non-ampere pip installation command, that performed a successful installation.

Collecting unsloth[cu118_ampere]@ git+https://github.com/unslothai/unsloth.git
  Cloning https://github.com/unslothai/unsloth.git to /tmp/pip-install-airv06iv/unsloth_a06586211bd7445ba155bf97e1192880
  Running command git clone --filter=blob:none --quiet https://github.com/unslothai/unsloth.git /tmp/pip-install-airv06iv/unsloth_a06586211bd7445ba155bf97e1192880
  Resolved https://github.com/unslothai/unsloth.git to commit 27bbd6b2cd927b7cb0866a03dec41efb04470501
  Installing build dependencies ... done
  Getting requirements to build wheel ... done
  Installing backend dependencies ... done
  Preparing metadata (pyproject.toml) ... done
WARNING: unsloth 2023.12 does not provide the extra 'cu118-ampere'
Collecting bitsandbytes (from unsloth[cu118_ampere]@ git+https://github.com/unslothai/unsloth.git)
  Using cached bitsandbytes-0.41.3.post2-py3-none-any.whl.metadata (9.8 kB)
Collecting flash-attn (from unsloth[cu118_ampere]@ git+https://github.com/unslothai/unsloth.git)
  Using cached flash_attn-2.3.6.tar.gz (2.3 MB)
  Installing build dependencies ... done
  Getting requirements to build wheel ... error
  error: subprocess-exited-with-error
  
  × Getting requirements to build wheel did not run successfully.
  │ exit code: 1
  ╰─> [17 lines of output]
      Traceback (most recent call last):
        File "/home/zmx/m.2/Dev/llm/unsloth/venv/lib/python3.10/site-packages/pip/_vendor/pyproject_hooks/_in_process/_in_process.py", line 353, in <module>
          main()
        File "/home/zmx/m.2/Dev/llm/unsloth/venv/lib/python3.10/site-packages/pip/_vendor/pyproject_hooks/_in_process/_in_process.py", line 335, in main
          json_out['return_val'] = hook(**hook_input['kwargs'])
        File "/home/zmx/m.2/Dev/llm/unsloth/venv/lib/python3.10/site-packages/pip/_vendor/pyproject_hooks/_in_process/_in_process.py", line 118, in get_requires_for_build_wheel
          return hook(config_settings)
        File "/tmp/pip-build-env-1h3yu7xe/overlay/lib/python3.10/site-packages/setuptools/build_meta.py", line 325, in get_requires_for_build_wheel
          return self._get_build_requires(config_settings, requirements=['wheel'])
        File "/tmp/pip-build-env-1h3yu7xe/overlay/lib/python3.10/site-packages/setuptools/build_meta.py", line 295, in _get_build_requires
          self.run_setup()
        File "/tmp/pip-build-env-1h3yu7xe/overlay/lib/python3.10/site-packages/setuptools/build_meta.py", line 480, in run_setup
          super(_BuildMetaLegacyBackend, self).run_setup(setup_script=setup_script)
        File "/tmp/pip-build-env-1h3yu7xe/overlay/lib/python3.10/site-packages/setuptools/build_meta.py", line 311, in run_setup
          exec(code, locals())
        File "<string>", line 9, in <module>
      ModuleNotFoundError: No module named 'packaging'
      [end of output]
  
  note: This error originates from a subprocess, and is likely not a problem with pip.
error: subprocess-exited-with-error

× Getting requirements to build wheel did not run successfully.
│ exit code: 1
╰─> See above for output.

note: This error originates from a subprocess, and is likely not a problem with pip.

Any idea what the cause might be?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions