Docker 中快速安装tensorflow环境

Docker 中快速安装tensorflow环境,并使用TensorFlow。

1、下载TensorFlow镜像

docker pull tensorflow/tensorflow

2、 建立TensorFlow容器

docker run --name corwien-tensortflow -it -p 8888:8888 -v /Users/kaiyiwang/Code/ai/notebooks:/notebooks/data  tensorflow/tensorflow

命令说明

  • docker run 运行镜像,
  • --name 为容器建立别名,
  • -it 保留命令行运行,
  • -p 8888:8888 将本地的8888端口 http://localhost:8888/ 映射,
  • -v /Users/kaiyiwang/Code/ai/notebooks:/notebooks/data 将本地的/Users/kaiyiwang/Code/ai/notebooks文件夹挂载到新建容器的/notebooks/data下(这样建立的文件能够保存到本地/Users/kaiyiwang/Code/ai/notebooks
  • tensorflow/tensorflow 为指定的镜像,默认标签为latest(即tensorflow/tensorflow:latest)

执行上边的命令:docker

clipboard.png

咱们能够看到,建立了TensorFlow容器,并给了一个默认登陆JupiterNotebook的页面。编程

咱们能够经过下面的命令在新的命令窗口看正在执行的容器,及容器所对应的映射端口数组

docker ps

clipboard.png

3、开启TensorFlow容器

1.能够直接从命令行中右键打开链接,或者在浏览器中输入http://127.0.0.1:8888,而后将命令行中的token粘贴上去。浏览器

clipboard.png

clipboard.png

4、开始TensorFlow编程

一、点击登陆进去能够看到界面了,而且能够new一个项目spa

clipboard.png

二、tensorflow示例源码解读命令行

from __future__ import print_function
#导入tensorflow
import tensorflow as tf
#输入两个数组,input1和input2而后相加,输出结果
with tf.Session():
    input1 = tf.constant([1.0, 1.0, 1.0, 1.0])
    input2 = tf.constant([2.0, 2.0, 2.0, 2.0])
    output = tf.add(input1, input2)
    result = output.eval()
    print("result: ", result)

三、运行程序,输出的结果为(运行成功)code

result:  [ 3.  3.  3.  3.]

5、相关命令

一、关闭或开启TensorFlow环境

#关闭tensorflow容器
docker stop corwien-tensortflow

#开启TensorFlow容器
docker start corwien-tensortflow
#浏览器中输入 http://localhost:8888/

二、文件的读写权限修改

#查看读写权限
ls -l
#将tensorflow 变为属于corwien(系统默认)用户
sudo chown -R corwien tensorflow/
#将tensorflow 变为属于corwien(系统默认)用户组
sudo chgrp -R corwien tensorflow/
相关文章
相关标签/搜索