Comment on Installing TensorFlow with Python 3 on EC2 GPU InstancesparentComments−dwiel10yIs running the docker image on a fresh standard AMI [1] all it takes to get a working tensorflow backed by the GPU? There is nothing you need to install on the host OS?[1] for example: Ubuntu 14.04 (HVM) public ami, ami-06116566−exxo_10yYou need the NVIDIA drivers and the nvidia-docker plugin.$ docker-machine create --driver amazonec2 --amazonec2-instance-type g2.2xlarge ...$ docker-machine ssh <host> # install the NVIDIA driver and nvidia-docker-plugin$ eval `docker-machine env <host>`$ ssh-add ~/.docker/machine/machines/<host>/id_rsa$ NV_HOST="ssh://ubuntu@<ip>:" nvidia-docker run mybuild/tensorflowStep 2 can be skipped if you use a custom AMI.
Comments
Is running the docker image on a fresh standard AMI [1] all it takes to get a working tensorflow backed by the GPU? There is nothing you need to install on the host OS?
[1] for example: Ubuntu 14.04 (HVM) public ami, ami-06116566
You need the NVIDIA drivers and the nvidia-docker plugin.
$ docker-machine create --driver amazonec2 --amazonec2-instance-type g2.2xlarge ...
$ docker-machine ssh <host> # install the NVIDIA driver and nvidia-docker-plugin
$ eval `docker-machine env <host>`
$ ssh-add ~/.docker/machine/machines/<host>/id_rsa
$ NV_HOST="ssh://ubuntu@<ip>:" nvidia-docker run mybuild/tensorflow
Step 2 can be skipped if you use a custom AMI.