This bash script exports all the CUDA libraries in Ubuntu/Linux environment for correct visibility of GPU devices. This is especially useful when using Tensorflow (> 3.15) in a Conda environment, which is not a default library location.

tf_env_variables.sh

# HOW TO RUN IN TERMINAL: 
# $ source tf_env_variables.sh

NVIDIA_DIR=$(dirname $(dirname $(python -c "import nvidia.cudnn;print(nvidia.cudnn.__file__)")))
for dir in $NVIDIA_DIR/*; do
    if [ -d "$dir/lib" ]; then
        export LD_LIBRARY_PATH="$dir/lib:$LD_LIBRARY_PATH"
    fi
done