1:首先根据自己需要使用的tensorflow-gpu版本来确定需要安装的cuda以及cudnn版本。
最新显卡 cuda cudnn匹配地址https://github.com/fo40225/tensorflow-windows-wheel
2:假如我们需要安装1.6.0的gpu,那么我们就需要先安装cuda 9.0 以及 cudnn 7.0
3:cuda下载地址:https://developer.nvidia.com/cuda-90-download-archive
sudo chmod +x cuda_9.0.176_384.81_linux.run # 为 cuda_9.0.176_384.81_linux.run 添加可执行权限
./cuda_9.0.176_384.81_linux.run # 安装 cuda_9.0.176_384.81_linux.run
修改 /etc/profile 文件
sudo vim /etc/profile
在末尾加上
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64$LD_LIBRARY_PATH
export CUDA_HOME=/usr/local/cuda
再source一下就可以了
source /etc/profile
sudo ldconfig
nvcc --version可以查看cuda的版本
4:cudnn下载地址:https://developer.nvidia.com/rdp/cudnn-download
sudo cp cuda/include/cudnn.h /usr/local/cuda-9.0/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-9.0/lib64
sudo chmod a+r /usr/local/cuda-9.0/include/cudnn.h /usr/local/cuda-9.0/lib64/libcudnn*