search

pytorch / ao

PyTorch native quantization and sparsity for training and inference
BSD 3-Clause "New" or "Revised" License
725 stars 92 forks source link

Installation Problems #788

Closed WAS-PlaiLabs closed 6 days ago

WAS-PlaiLabs commented 2 weeks ago

I cannot install this properly. Pypi is upgraded to 24.2, and says it's satisfied, but the only version that exist on my pip are 0.0.1, 0.0.3, and 0.1.

For example, running nightly install: !pip install --pre --force-reinstall torchao --index-url https://download.pytorch.org/whl/nightly/cu121

Results in: Successfully installed MarkupSafe-2.1.5 filelock-3.13.1 fsspec-2024.6.1 jinja2-3.1.4 mpmath-1.3.0 networkx-3.3 numpy-1.26.4 packaging-22.0 sentencepiece-0.2.0 sympy-1.13.1 torch-2.5.0.dev20240901+cu121 torchao-0.1 typing-extensions-4.12.2

Where only torchao-0.1 is installed

torchao (0.1)
Available versions: 0.1, 0.0.3, 0.0.1
  INSTALLED: 0.1
  LATEST:    0.1
msaroufim commented 2 weeks ago

What operating system are you using? If it's not Linux, you'll need to install ao from source

WAS-PlaiLabs commented 2 weeks ago

That's unfortunate, but good to know.

Let's see if I still have strange issues with WSL2! I can't dual boot with a compatible Ubuntu for some reason either cause 4090 drivers cause kernel to die if I ever restart the machine.

msaroufim commented 6 days ago

@WAS-PlaiLabs did WSL end up working for you?

WASasquatch commented 6 days ago

Yes, I did get it working on WSL2, thank you.