官术网_书友最值得收藏!

Setup using Docker

The previous section describes getting started from scratch which can be tricky sometimes given continuous changes to software packages and changing links on the web. One way to avoid dependence on links is to use container technology like Docker.

In this chapter, we will use the official NVIDIA-Docker image that comes pre-packaged with all the necessary packages and deep learning framework to get you quickly started with deep learning application development:

$ sudo add-apt-repository ppa:graphics-drivers/ppa -y
$ sudo apt-get update
$ sudo apt-get install -y nvidia-375 nvidia-settings nvidia-modprobe
  1. We now install Docker Community Edition as follows:
$ curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo apt-key add -
# Verify that the key fingerprint is 9DC8 5822 9FC7 DD38 854A E2D8 8D81 803C 0EBF CD88
$ sudo apt-key fingerprint 0EBFCD88
$ sudo add-apt-repository \
"deb [arch=amd64] https://download.docker.com/linux/ubuntu \
$(lsb_release -cs) \
stable"
$ sudo apt-get update
$ sudo apt-get install -y docker-ce
  1. We then install NVIDIA-Docker and its plugin:
$ wget -P /tmp https://github.com/NVIDIA/nvidia-docker/releases/download/v1.0.1/nvidia-docker_1.0.1-1_amd64.deb
$ sudo dpkg -i /tmp/nvidia-docker_1.0.1-1_amd64.deb && rm /tmp/nvidia-docker_1.0.1-1_amd64.deb
  1. To validate if the installation happened correctly, we use the following command:
$ sudo nvidia-docker run --rm nvidia/cuda nvidia-smi
  1. Once it’s setup correctly, we can use the official TensorFlow or Theano Docker image:
$ sudo nvidia-docker run -it tensorflow/tensorflow:latest-gpu bash
  1. We can run a simple Python program to check if TensorFlow works properly:
import tensorflow as tf
a = tf.constant(5, tf.float32)
b = tf.constant(5, tf.float32)
with tf.Session() as sess:
sess.run(tf.add(a, b)) # output is 10.0
print("Output of graph computation is = ",output)

You should see the TensorFlow output on the screen now as shown in figure Tensorflow sample output:

Tensorflow sample output
主站蜘蛛池模板: 漳浦县| 和林格尔县| 保靖县| 囊谦县| 海原县| 游戏| 神池县| 贵港市| 友谊县| 扎赉特旗| 呼图壁县| 象州县| 武平县| 元阳县| 穆棱市| 淳化县| 澄迈县| 乌拉特后旗| 宝兴县| 扎兰屯市| 乐昌市| 宝坻区| 醴陵市| 碌曲县| 福建省| 六盘水市| 高清| 盐城市| 乃东县| 唐海县| 和平区| 磴口县| 望江县| 宁化县| 颍上县| 东平县| 建宁县| 汪清县| 奉贤区| 嘉定区| 诸暨市|