linux 安装tensorflow(gpu版本)

一、安装cuda

  1.在nvidia官网下载CUDA  (linux版本cuda8.0)下载.run文件  下载好后的cuda_8.0.61_375.26_linux.run安装文件移动至当前用户根目录下 终端执行:

    linux 安装tensorflow(gpu版本)

2.在安装界面中,可参照下面的描述进行选择
linux 安装tensorflow(gpu版本)

3.添加环境变量,安装结束后,在管理员用户根目录下,找到.bashrc文件并打开,在最后添加下面三行文本,保存退出即可

linux 安装tensorflow(gpu版本)

4.查看duda是否安装好

    重启终端后,执行nvcc -V,若显示以下信息,则安装cuda成功

linux 安装tensorflow(gpu版本)

二、安装tensorflow

1. 安装bcupti-dev library可执行

     sudo apt-get install libcupti-dev

2.用python直接安装

  首先需要pip,如果没有的话可以使用下面的命令安装
sudo apt-get install python-pip python-dev   # for Python 2.7
sudo apt-get install python3-pip python3-dev # for Python 3.n

  下面使用pip安装tensorflow

$ sudo pip install tensorflow # Python 2.7; CPU support (no GPU support)

$ sudo pip3 install tensorflow # Python 3.n; CPU support (no GPU support)

$ sudo pip install tensorflow-gpu # Python 2.7; GPU support

$ sudo pip3 install tensorflow-gpu # Python 3.n; GPU support

如: sudo pip install tensorflow-gpu ==1.4

三、测试tensorflow是否安装成功

经过一二两个步骤,应该是安装成功了,可写个tensorflow的小程序测试一下

linux 安装tensorflow(gpu版本)