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

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
主站蜘蛛池模板: 巴塘县| 方正县| 罗江县| 遂昌县| 东阿县| 松潘县| 弥渡县| 隆子县| 和平县| 深圳市| 安吉县| 比如县| 巴林右旗| 塘沽区| 广元市| 岢岚县| 贵德县| 饶河县| 临沧市| 砀山县| 绵阳市| 拜城县| 安国市| 旌德县| 钟山县| 瑞金市| 淮滨县| 屯门区| 西城区| 宁强县| 自贡市| 三河市| 红河县| 温宿县| 柯坪县| 嘉义县| 肥乡县| 政和县| 新邵县| 玛沁县| 伊吾县|