TensorFlow安装+入门操作

1.创建专门运行环境,打开终端模拟器,为Anaconda创建一个python3.5(或者3.6都行,随便)的环境,环境名称为tensorflow ,在命令行中输入下面命令:

view plain copy

  1. conda create -n tensorflow   

2.启动tensorflow的环境:

view plain copy

  1. source activate tensorflow  

注:当不使用tensorflow时,关闭tensorflow环境,命令为:

view plain copy

  1. source deactivate tensorflow  

3.安装cpu版本的TensorFlow

view plain copy

  1. sudo pip install --upgrade --ignore-installed tensorflow  

测试是否安装完成

4.在命令行输入Python

view plain copy

  1. python  

 

5.在Python环境,使用import导入TensorFlow模块,别名为tf,Python代码中指定GPU。

view plain copy

  1. import tensorflow as tf  
  2. import os  
  3. os.environ["CUDA_VISIBLE_DEVICES"]="0"  

6.构造计算图,创建一个常量节点hello,值为Hello,TensorFlow!,代码如下:

view plain copy

  1. hello = tf.constant('Hello, TensorFlow!')  

7.创建一个Session会话对象,设置定量的GPU使用量,会话封装了Tensorflow运行时的状态和控制。

view plain copy

  1. config = tf.ConfigProto()   
  2. config.gpu_options.per_process_gpu_memory_fraction = 0.9 # 占用GPU90%的显存   
  3. sess= tf.Session(config=config)  

8.运行计算图,调用run方法,运行计算图。

view plain copy

  1. print(sess.run(hello))  

 

到此在Linux的anaconda环境下,TensorFlow安装测试完成!

 

posted @ 2021-05-07 17:20  Spongbob  阅读(141)  评论(0编辑  收藏  举报