Did the install logs show that the cudatoolkit was indeed installed or did you see a fallback to a CPU version?
If you’ve verified that the CUDA version was indeed picked, the error might be raised by a setup issue, i.e. your GPU cannot be detected.
Is nvidia-smi working properly and do you have a sufficiently new NVIDIA driver to run the CUDA11.1 runtime?