This images helps to train neural networks on a GPU without installing Python packages on host
807
You simply need to install NVIDIA drivers. The image will take care of the rest
Python environment already includes tensorflow-gpu, kearas, and some other basic ML libraries
Command line parameters to start jupyter:
docker run --gpus all -it -p 8889:8889 -v /home/ubuntolog/data:/data ubuntolog/tensorflow-gpu
Content type
Image
Digest
Size
1.7 GB
Last updated
over 6 years ago
docker pull ubuntolog/tensorflow-gpu