Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add Docker image, Makefile and requirements.txt #7

Open
wants to merge 6 commits into
base: master
Choose a base branch
from

Conversation

juniorxsound
Copy link

Why?
Adding a Docker image based on the Tensorflow Docker image to create the installation and setup process easy and breezy.

What?
Adding Dockerfile.gpu which can easily be called using a Makefile that has the following commands:

  • make build build the docker container
  • make example run the pretained_example.py in the container
  • make shell open bash inside the container
  • make jupyter to run jupyter notebooks inside the container (would love to add notebook examples in a separate PR)

The setup uses the latest version of nvidia-docker and requires Docker 19.03 or newer

@frankcarey
Copy link

There is an error if you try to use the make jupyter command after make build because there is no ./notebooks directory. You need to either create it using mkdir notebooks or this PR could ship with a README.md in a new ./notebooks directory?

@juniorxsound
Copy link
Author

Thanks @frankcarey, I have a bunch of example notebooks but feel like it might be a better idea to separate the examples into a PR since it might grow to be a larger change. I'll get rid of make jupyter for this PR and add to the next one

…-w /data --rm -it -v `pwd`:/data -t "juniorxsound/stylegan:latest" jupyter notebook ./notebooks --ip=0.0.0.0 --allow-root \)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants