This introduction is using tensorflow1.8, cuda9.0, cudnn7.0 version
- https://developer.nvidia.com/cuda-90-download-archive download the appropriate cuda, it is recommended to install with Deb
sudo dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64.debsudo apt-key add /var/cuda-9-0-local/7fa2af80.pubsudo apt-get updatesudo apt-get install cuda
- HTTPS://DEVELOPER.NVIDIA.COM/CUDNN Download CUDNN
tar -xzvf cudnn-9.0-linux-x64-v7.tgzsudo cp cuda/include/cudnn.h /usr/local/cuda/includesudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
- Configure the environment variables, add the following two sentences to the ~/.BASHRC
export PATH=/usr/local/cuda/bin:$PATHexport LD_LIBRARY_PATH=/usr/local/cuda/lib64$LD_LIBRARY_PATH
- Install TENSORFLOWW-GPU, and Nvidia library is finished.
sudo apt-get install nvidia-cuda-toolkitpip install tensorflow-gpu
ubuntu16.04 CUDA, CUDNN installation