I added a Dockerfile that creates an image based on nvidia CUDA 9.0 + CUDNN 7. The image contains:
Miniconda python 3
Required python dependencies for pytorch and Jupyter Notebook.
The code from this repository
The image is configured so that a Jupyter Notebook is run at start time.
An enviroment.yml file is also included, stating all the required python dependencies. This could be useful as well for users who still want to run the code on the host but prefer to install everything in an isolated python environment.
The README.md has also been updated with instructions on how to build and run the Docker image.
DmitryUlyanov
commented
6 years ago
Solves #3
I added a Dockerfile that creates an image based on nvidia CUDA 9.0 + CUDNN 7. The image contains:
The image is configured so that a Jupyter Notebook is run at start time.
An enviroment.yml file is also included, stating all the required python dependencies. This could be useful as well for users who still want to run the code on the host but prefer to install everything in an isolated python environment.
The README.md has also been updated with instructions on how to build and run the Docker image.