在ubuntu16.04 安装 tensorflow-gpu

一. 需要的环境和软件:

  • [Ubuntu 16.04](https://www.ubuntu.com/download/desktop,"ubuntu 下载地址")
  • Anaconda 3
  • Cuda-8.0
  • Cudnn5.1
  • Tensorflow-gpu

二. 安装步骤:

  1. 查看显卡信息 sudo lshw -numeric -C display
  2. 安装Nvidia 显卡驱动,建议直接在ubuntu 设置中跟新设定为NVIDIA私有驱动
  3. 安装Cuda8.0
  • export PATH=$PATH:/usr/local/cuda-8.0/bin
  • export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64
  • nvidia-smi

4.下载cudnn5.1,并复制到 cuda 安装目录的相应文件夹下
5 安装tensorflow

你可能感兴趣的:(在ubuntu16.04 安装 tensorflow-gpu)