CUDA not available
RaymondTracer opened this issue · 1 comments
RaymondTracer commented
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
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/