geezlab/cuda-tensorflow-pytorch-notebook repository overview

Please use: geezlab/gpu-deep-learning-notebook⁠

⁠Content

• Nvidia CUDA 9.0
• TensorFlow
• PyTorch
• NLP (spaCy, nltk, torchtext)
• Python machine learning Libraries

⁠Pull Container Image

This will take a while depending on your internet connection.

docker pull geezlab/cuda-tensorflow-pytorch-notebook

⁠Run Container

Create an ephemeral container named dl-app, which will mount your current working directory to ~/dev within the container. Note that this container will be destroyed once Jupyter is stopped if you don't like this approach replace --rm with -d in the following command.

docker run --rm -p 10000:8888 -e JUPYTER_ENABLE_LAB=yes --runtime="nvidia" -v "$PWD":/home/geez/dev --name dl-app cuda-tensorflow-pytorch-notebook:latest

This will print a full URL with token that can be use to browse Jupyter.

⁠Jupyter in Browser

`http://<host-ip>:10000/?token=<token>`

Happy Coding!