cuda cudnn tensorflow-gpu安装使用教程

1：首先根据自己需要使用的tensorflow-gpu版本来确定需要安装的cuda以及cudnn版本。

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*

扫描二维码关注公众号，回复： 11911242 查看本文章