openai / shap-e

Generate 3D objects conditioned on text or images
MIT License
11.54k stars 928 forks source link

Stuck at 0% #64

Open ShacharWeis opened 1 year ago

ShacharWeis commented 1 year ago

I cloned the repo and installed the packages needed. Running on a AMD Ryzen 9 laptop with an RTX3070 at 16gb RAM

I run a text-to-model sample and I see this: 0%| | 0/64 [00:00<?, ?it/s]

after 15 minutes it's still at 0% Is it stuck or just really slow?

Kabanosk commented 1 year ago

I think that is because you run computation on CPU instead of GPU. Can you print the value of thedevice variable?

ShacharWeis commented 1 year ago

Yes, that was the problem. I installed torch with CUDA and now it works.

Here is the fix:

pip installing torch pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

BryanValc commented 1 year ago

Yes, that was the problem. I installed torch with CUDA and now it works.

Here is the fix:

pip installing torch pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

For me it still doesn't work, I have an RTX 3060 12G and Ryzen 5 2600, isn't it because of the python version? I'm using Python 3.11.3, let me guys know what version are you using

Kabanosk commented 1 year ago

@BryanValc Can you please write what is value of device variable?

device = 'cuda' if torch.cuda.is_available() else 'cpu'
print(device)
BryanValc commented 1 year ago

@Kabanosk I did that, but it keeps using cpu image

ShacharWeis commented 1 year ago

You have to uninstall torch and then install a CUDA version of torch. Then the variable value is "cuda"

Or, you can just not bother, because the 3D models output from this AI are complete garbage. Really. The examples they show here were carefully hand picked. It's a waste of time (in it's current state). I played with it for an hour and gave up. Nothing it makes is even remotely unusable.

BryanValc commented 1 year ago

You have to uninstall torch and then install a CUDA version of torch. Then the variable value is "cuda"

Or, you can just not bother, because the 3D models output from this AI are complete garbage. Really. The examples they show here were carefully hand picked. It's a waste of time (in it's current state). I played with it for an hour and gave up. Nothing it makes is even remotely unusable.

I tried uninstalling it, then running this command: pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

But still nothing.. maybe you're right, it isn't much of a big deal as of now, it is like the beginning of stable diffusion

scottboyd-ai commented 1 year ago

You may need to delete your vevn folder or create a new venv. If the requirement is already satisfied according to pip, it won't install the cuda version.

Le0m1 commented 8 months ago

123 i have a problem with GPU, but i can't fix this

siddharth952 commented 4 months ago

pip uninstall torch pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 Then restart your kernel