Running Tensorflow with Nvidia Docker
In this post you’ll learn how to use nvidia-docker with Tensorflow.
Pulling the Tensorflow GPU Image
Use the correct image based on your CUDA version:
- CUDA 8.0 use
tensorflow/tensorflow:latest-gpu
- CUDA 7.5 use
tensorflow/tensorflow:1.0.0-rc0-gpu
Test nvidia-smi
nvidia-docker run --rm tensorflow/tensorflow:latest-gpu nvidia-smi
That’s it! Now your ready to run your Tensorflow code.
Example Dockerized Tensorflow Project using GPU
1) Create a Dockerfile
2) Create app.py
3) Build a running nvidia-docker ⤧ Next post Training on Large Scale Image Datasets with Keras