3 ms·

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

by dwiel 11y ago

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

exxo_ 11y ago
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.