Problems:
Errors as follows:
RuntimeError: CUDA error: no kernel image is available for execution on the device
CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1.
Solutions:
Reinstall pytorch. Please reference the torch version according to the homepage of pytorch. pytorch homepage
pip uninstall torch
pip uninstall torchaudio
pip uninstall torchvision
pip3 install torch torchvision torchaudio