Open philipbutler opened 1 month ago
What cuda version do you have installed?
You'll need to install the right one from
pip install torchao --extra-index-url https://download.pytorch.org/whl/cu121 # full options are cpu/cu118/cu121/cu124
Oh apparently I had 12.2. Would that be an issue?
Yeah I'd recommend you still everything in a fresh conda environment and try again with the cuda versions that match exactly
Otherwise installing ao from source should always work
I was running a very simple pytorch program that worked fine before, but after
pip install torchao
, I encounteredI tried uninstalling pytorch then installing the nightly (Linux [Ubuntu 24.04]. CUDA 12.2), and was met with a different error
and lastly I did
pip uninstall torchao
but still have the same error as above :(