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

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
主站蜘蛛池模板: 东乡族自治县| 石阡县| 新密市| 明溪县| 柳州市| 宜宾市| 海阳市| 咸宁市| 巴东县| 玉龙| 垫江县| 临武县| 兴和县| 喀喇| 信丰县| 新绛县| 郸城县| 长海县| 霞浦县| 吴忠市| 始兴县| 屯门区| 舒兰市| 望城县| 峨山| 临汾市| 增城市| 湘潭市| 自治县| 张掖市| 肥乡县| 浦城县| 定南县| 贺兰县| 刚察县| 乌恰县| 阳原县| 麦盖提县| 涿鹿县| 兴海县| 银川市|