ubuntu 20.04 tensorflow nvidia cuda cudnn

nvidia-driver

sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/7fa2af80.pubsudo add-apt-repository "deb http://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/ /"sudo apt-get updatesudo apt-get -y install cuda

if there is something wrong with the 20 .04 apt repository,  please change the ubuntu2004 into 1810,such as

https://developer.download.nvidia.cn/compute/cuda/repos/ubuntu1810/x86_64/

sudo apt install nvidia-driver-440

dowload cudnn from https://developer.nvidia.com/rdp/cudnn-download

and untared the arch

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64
export PATH=/usr/local/cuda/bin:$PATH

pip install -U --user pip six 'numpy<1.19.0' wheel setuptools mock 'future>=0.17.1'pip install -U --user keras_applications --no-depspip install -U --user keras_preprocessing --no-depssudo apt install bazel

git clone https://github.com/tensorflow/tensorflow.gitcd tensorflowgit checkout branch_name  # r1.9, r1.10, etc.
./configure

bazel build --config=opt --config=cuda //tensorflow/tools/pip_package:build_pip_package  

maybe the version of gcc is too high, and type 'sudo apt install gcc-7 g++-7' works. the gcc-7 location is /usr/bin/gcc-7