CUDA not available - Githubissues

kuprel / min-dalle

min(DALL·E) is a fast, minimal port of DALL·E Mini to PyTorch

MIT License

3.48k stars 255 forks source link

CUDA not available #66

Closed RaymondTracer closed 2 years ago

RaymondTracer commented 2 years ago

When I run image_from_text.py I get an error saying "User provided device_type of 'cuda', but CUDA is not available. Disabling" in autocast_mode.py

I have an RTX 3070 Ti so CUDA should work.

superswan commented 2 years ago

For CUDA support on either windows or linux you will need to first install the CUDA runtime. apt install cuda for linux Google for the windows binary

It is also likely you only installed the cpu version of PyTorch. You need to specify CUDA.

The following command will install PyTorch built with CUDA support for Windows pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113

Source: https://pytorch.org/