System information

• OS Platform and Distribution (e.g., Linux Ubuntu 18.04): Ubuntu 20.04
• Python version: 3.6
• Installed using virtualenv
• CUDA/cuDNN version: 11.5 / 8.1.0.77
• GPU model and memory: RTX 3090 24GB nvidia driver 460.39
• TensorFlow version: 2.4.0 pip install tensorflow-gpu==2.4.0

Describe the problem

Installed cuda 11.2 and cudnn 8.1.0.77. Faced the following problem when I run train.py

Could not load dynamic library 'libcusolver.so.10'; dlerror: libcusolver.so.10: cannot open shared object file

Solved the problem

• place libcusolver.so.11 in the tensorflow site-packages of the virtual environment created worked properly
• followed the link to solve the problem

https://github.com/tensorflow/tensorflow/issues/43947#issuecomment-722019317

• Manage a link between libcusolver.so.11 and libcusolver.so.10 using a comand 'sudo ln -s'

$ ln -s /usr/local/cuda/lib64/libcusolver.so.11 ~.virtualenvs/(env name)/lib/python3.6/site-packages/tensorflow/python/libcusolver.so.10

This fixed the problem for me