1. download ubuntu: https://cn.ubuntu.com/download/desktop
2. follow instructions to install ubuntu:Install Ubuntu desktop | Ubuntu
3.install docker engine: Install Docker Engine on Ubuntu | Docker Docs
4.pull tensorflow docker iamges: Docker | TensorFlow
5.install nvidia-docker-toolkit:
GitHub - NVIDIA/nvidia-docker: Build and run Docker containers leveraging NVIDIA GPUsInstalling the NVIDIA Container Toolkit — NVIDIA Container Toolkit 1.14.2 documentation
6.test your tensorflow-gpu-jupyter
- 1. test on bash:
- docker run --gpus all -it --rm tensorflow/tensorflow:latest-gpu \
- python -c "import tensorflow as tf;import numpy as np;import time;value =np.random.randn(5000,1000);a = tf.constant(value);tic = time.time();b = [tf.reduce_sum(a*a) for i in range(1000)];toc = time.time();t_cost = toc - tic;print(t_cost);"
-
-
-
-
- 2. test on notebook
- 2.1 gpu version
- docker run --gpus all -p 8888:8888 tensorflow/tensorflow:latest-gpu-jupyte
-
- 2.2 cpu version
- docker run -it -p 8888:8888 tensorflow/tensorflow:latest-gpu-jupyter
-
- 2.3 test time used
- import tensorflow as tf;
- import numpy as np;
- import time;
- value = np.random.randn(5000,1000);
- a = tf.constant(value);
- tic = time.time();
- b = [tf.reduce_sum(a*a) for i in range(1000)];
- toc = time.time();
- t_cost = toc - tic;
- print(t_cost);
-
-