search

lllyasviel / Paints-UNDO

Understand Human Behavior to Align True Needs
Apache License 2.0
3.25k stars 287 forks source link

Torch not compiled with CUDA enabled #6

Open barepixels opened 1 month ago

barepixels commented 1 month ago

Paint by numbers on steroids!!!

noob with 3090 need help. I followed instructions how to install but getting "Torch not compiled with CUDA enabled"

git clone https://github.com/lllyasviel/Paints-UNDO.git
cd Paints-UNDO
conda create -n paints_undo python=3.10
conda activate paints_undo
pip install xformers
pip install -r requirements.txt
python gradio_app.py

can someone provide CMD line to replace with proper Torch with CUDA? or how to edited requirements.txt?

jtydhr88 commented 1 month ago

on windows, you should install torch first, then requirements.txt: pip install torch==2.3.0 torchvision torchaudio xformers --index-url https://download.pytorch.org/whl/cu121

barepixels commented 1 month ago

Thanks jtydhr88 Got it working

shiverwang76 commented 1 month ago

on windows, you should install torch first, then requirements.txt: pip install torch==2.3.0 torchvision torchaudio xformers --index-url https://download.pytorch.org/whl/cu121

It works, thanks