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 'libcupti.so.11.0'; dlerror: libcupti.so.11.0: cannot open shared object file