TensorFlow安装+入门操作
1.创建专门运行环境,打开终端模拟器,为Anaconda创建一个python3.5(或者3.6都行,随便)的环境,环境名称为tensorflow ,在命令行中输入下面命令:
- conda create -n tensorflow
2.启动tensorflow的环境:
- source activate tensorflow
注:当不使用tensorflow时,关闭tensorflow环境,命令为:
- source deactivate tensorflow
3.安装cpu版本的TensorFlow
- sudo pip install --upgrade --ignore-installed tensorflow
测试是否安装完成
4.在命令行输入Python
- python
5.在Python环境,使用import导入TensorFlow模块,别名为tf,Python代码中指定GPU。
- import tensorflow as tf
- import os
- os.environ["CUDA_VISIBLE_DEVICES"]="0"
6.构造计算图,创建一个常量节点hello,值为Hello,TensorFlow!,代码如下:
- hello = tf.constant('Hello, TensorFlow!')
7.创建一个Session会话对象,设置定量的GPU使用量,会话封装了Tensorflow运行时的状态和控制。
- config = tf.ConfigProto()
- config.gpu_options.per_process_gpu_memory_fraction = 0.9 # 占用GPU90%的显存
- sess= tf.Session(config=config)
8.运行计算图,调用run方法,运行计算图。
- print(sess.run(hello))
到此在Linux的anaconda环境下,TensorFlow安装测试完成!