Skip to content

AVIATR/tf_jupyter

Repository files navigation

tf_jupyter

Docker container for machine learning, runs jupyter notebook & tensorflow

Building

Simply clone the repo, and run ./build.sh to build a container that is automatically tagged with the branch name. To also build the gpu version, run ./build.sh -g. The GPU version is built from Dockerfile_gpu.

The automatic builds for branch master can be found on Dockerhub as aviatr/tf_jupyter:latest and aviatr/tf_jupyter:gpu. For other branches, the auto-builds are tagged as aviatr/tf_jupyter:<branchname> and aviatr/tf_jupyter:gpu-<branchname>.

Using the containers

By default, the container runs jupyter notebook on port 8888, and can be launched simply by running the ./run.sh script. To use the nvidia-gpu runtime, you can pass the -g option, and to use a different tag other than master, you can use the -t <tagname> option. For more info, run ./run.sh -h.

If you would like to put this in the cloud somewhere, create a private folder, and replace the sha-1 hashed password in pwd.txt file (currently password) with your own password. Furthermore, add your ssl keys underneath private. You can then launch the container with the script ./run.sh which will mount the private folder in the docker container where the jupyter notebook expects to find them, and also passes your password to the jupyter notebook.

Acknowledgements:

Dark theme for Jupyter Notebook/iPython 4 is created by Theodore Pak.

About

Docker container with tensorflow and python

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published