鉴于pip安装tensorflow容易出现runtime error的问题,本文使用anaconda中的conda install安装tensorflow.
默认cuda和 cudnn已经安装好,本文从安装Anaconda开始。
1. 安装Anaconda
方法(1)
到 Anaconda download site下载并按照网页上的步骤安装Anaconda
方法(2)需要先下载
$ bash Anaconda2-4.3.1-Linux-x86_64.sh #Python 2.7版本
或者
$ bash Anaconda3-4.3.1-Linux-x86_64.sh #Python 3.5 版本
在安装的过程中,会询问安装路径,按回车即可。之后会询问是否将Anaconda安装路径加入到环境变量(.bashrc)中,输入yes,这样以后在终端中输入python即可直接进入Anaconda的Python版本(如果你的系统中之前安装过Python,自行选择yes or no)。安装成功后,会有当前用户根目录下生成一个anaconda2的文件夹,里面就是安装好的内容
NOTE:安装完成后需重启终端conda才能生效
查询安装信息
$ conda info
查询当前已经安装的库
$ condalist
安装库(***代表库名称)
$ conda install
更新库
$ conda update
2. 安装tensorflow(conda install安装)
到anaonda cloud tensorflow里面查资源和安装命令
3. 验证是否安装成功
python
>>>import tensorflow as tf
没有报错则安装成功
4. NOTE:
指定运行时用的GPU,先查看那个GPU空闲,然后再指定。
nvidia-smi
import os
os.environ["CUDA_VISIBLE_DEVICES"]='1'