Apply filter in docker container, supporting GPU #268
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
I added the file nut.yml, so that generating images in a docker image, on GPU, can be as easy as running the command
nut run
in the repo. This address issues/PR such as #151Nut is a tool to run commands in a docker container, while supporting nvidia devices, volumes, etc. Here, nut.yml declares the following settings:
matthieudelaro/cuda-torch-plus
/opt/style
Note that Docker supports GPUs on Linux only, and that Nut relies on nvidia-docker-plugin (details).
In case GPUs are not available, nut will display a warning message and run the neural network.
To run the neural network with a custom command, you can use
--exec
flag. For example: