Open cheesesteak45 opened 1 year ago
Same question here.
Not sure if this is related to xformer. I cannnot install the recommended xformer version.
note: Some user reports ValueError: fp16 mixed precision requires a GPU is occurred in training. In this case, answer 0 for the 6th question: What GPU(s) (by id) should be used for training on this machine as a comma-separated list? [all]:
(Single GPU with id 0 will be used.)
Maybe this could be helpful?
first, you need rocm enabled pytorch:
pip uninstall torch
pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/rocm5.4.2
second, there has to a rocm enabled xformers: yet TBD, in other words impossible: see https://github.com/facebookresearch/xformers/issues/485
Trying to train on an AMD RX6600XT on arch linux. Xformers is disabled, and in the startup script, when it prompts which GPU I want to use, I've tried entering "all", "[all]", "0", and "1", but none of them make the script recognize my GPU and I get
ValueError: fp16 mixed precision requires a GPU
every time. Any idea of what the issue could be?The full output is as follows: