Merge pull request #6505 from feiskyer/docker-setup

Update docker setup docs
This commit is contained in:
Shanqing Cai 2016-12-26 19:50:28 -05:00 committed by GitHub
commit 40dcfc6f92

View File

@ -500,14 +500,13 @@ For NVidia GPU support install latest NVidia drivers and
$ nvidia-docker run -it -p 8888:8888 gcr.io/tensorflow/tensorflow:latest-gpu
```
If you have a problem running `nvidia-docker`, then using the default config, we
include a
[script](https://github.com/tensorflow/tensorflow/blob/master/tensorflow/tools/docker/docker_run_gpu.sh)
in the repo with these flags, so the command-line would look like
Note: If you have a problem running `nvidia-docker`, you may try the old way
we have used. But it is not recomended. If you find a bug in `nvidia-docker`, report
it there please and try using the `nvidia-docker` as described above.
```bash
$ path/to/repo/tensorflow/tools/docker/docker_run_gpu.sh -p 8888:8888 gcr.io/tensorflow/tensorflow:latest-gpu
```
$ export CUDA_SO=$(\ls /usr/lib/x86_64-linux-gnu/libcuda.* | xargs -I{} echo '-v {}:{}')
$ export DEVICES=$(\ls /dev/nvidia* | xargs -I{} echo '--device {}:{}')
$ docker run -it -p 8888:8888 $CUDA_SO $DEVICES gcr.io/tensorflow/tensorflow:latest-gpu
For more details see [TensorFlow docker
readme](https://github.com/tensorflow/tensorflow/tree/master/tensorflow/tools/docker).