Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

virtual environment with torch cpu forces to install torch gpu #3317

Open
4 tasks
mgathena opened this issue Dec 27, 2024 · 0 comments
Open
4 tasks

virtual environment with torch cpu forces to install torch gpu #3317

mgathena opened this issue Dec 27, 2024 · 0 comments

Comments

@mgathena
Copy link

System Info

N/A
(reason: couldn't install)
Python 3.10.12

Information

  • The official example scripts
  • My own modified scripts

Tasks

  • One of the scripts in the examples/ folder of Accelerate or an officially supported no_trainer script in the examples folder of the transformers repo (such as run_no_trainer_glue.py)
  • My own task or dataset (give details below)

Reproduction

I am trying to install accelerate via uv. I already have torch==2.4.1+cpu installed. but when i install via uv add accelerate, it starts to download cuda/triton/nccl/torch(GPU). Is there a way to prevent the installation of GPU version of torch? I also tried to pin the version like uv add accelerate==0.14, but it didn't work.

Expected behavior

It shouldn't download the GPU version and continue to work with the CPU version of pytorch

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant