No CUDA GPU found in your device

[cchatgpt]

[yangheng95]

Please try input nvidia-smi in your cmd, if it does not work, you may need to fix cuda by reinstallation.

[yangheng95]

Please check if you install the pytorch with cuda bundle, use the following code:

import torch
print(torch.cuda.is_available())

If you install the cpu version, please reinstall pytorch from: https://pytorch.org/get-started/locally/

[yangheng95]

Yes if you use conda, otherwise if you use pip, you can use: pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116