Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Torch #26

Open
Xcelerate04 opened this issue Apr 11, 2023 · 2 comments
Open

Torch #26

Xcelerate04 opened this issue Apr 11, 2023 · 2 comments

Comments

@Xcelerate04
Copy link

how do you fix "AssertionError: Torch not compiled with CUDA enabled"
and " ModuleNotFoundError: No module named 'torch._six' "

@Tropdilee
Copy link

If you get a message like this:
File "C:\Users\Gameroom\ai-art-generator\taming-transformers\taming\data\utils.py", line 11, in
from torch._six import string_classes
ModuleNotFoundError: No module named 'torch._six'

Go to the location that is given and open the file with a text editor:
File "C:\Users\Gameroom\ai-art-generator\taming-transformers\taming\data\utils.py"

And put a # before the code:
In my case is it on line 11.
from torch._six import string_classes

Like:
#from torch._six import string_classes

For me it worked

@Tropdilee
Copy link

For "AssertionError: Torch not compiled with CUDA enabled" you must download a diffenent version of Pytorch at step #2.

this is given to you:
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

and for example you must have:
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

Use this link at step #2
Note that you can customize your Pytorch installation by using the online tool located here.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants