https://github.com/floydhub/dockerfiles

Deep Learning Dockerfiles
https://github.com/floydhub/dockerfiles

deep-learning docker pytorch tensorflow torch

Last synced: 3 months ago
JSON representation

Deep Learning Dockerfiles

• Host: GitHub
• URL: https://github.com/floydhub/dockerfiles
• Owner: floydhub
• License: apache-2.0
• Created: 2017-01-03T01:23:55.000Z (almost 8 years ago)
• Default Branch: master
• Last Pushed: 2021-04-20T17:51:45.000Z (over 3 years ago)
• Last Synced: 2024-07-19T02:15:15.943Z (4 months ago)
• Topics: deep-learning, docker, pytorch, tensorflow, torch
• Language: Python
• Homepage: https://docs.floydhub.com/guides/environments/
• Size: 533 KB
• Stars: 156
• Watchers: 7
• Forks: 57
• Open Issues: 9
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

        
# Dockerfiles

Collection of Dockerfiles useful for NLP and Deep Learning. To download the docker images

visit: [floydhub's Docker Hub](https://hub.docker.com/r/floydhub/).

## How to update framework

1. Dockerfiles are generated using two inputs: `matrix.yml` and jinja template file inside `./dl/FRAMEWORK` directory. `matrix.yml` provides variable values for jinja template files.

  * The `$render` list in `matrix.yml` controls what version of the framework to render.

  * For each version config in `matrix.yml`, any key starts with `_` are global keys, which will get automatically injected into each variant config for that version.

2. Most of the cases, you only need to update `./dl/FRAMEWORK/matrix.yml` to generate a set of dockerfiles for a new version of a framework. If not, you will need to update the jinja file to account for build step changes.

3. Install floydker: `cd floydker && pipenv shell && pipenv install`.

4. Render dockerfiles: `cd .. && floydker render .`.

5. Commit new docker images to git and push: `git commit -a`.

## Naming conventions

Dockerfiles are organized into the following directory structure:

```

CATEGORY/PROJECT_NAME/VERSION/Dockerfile-ENV

CATEGORY/PROJECT_NAME/VERSION/Dockerfile-ENV.gpu

```

Automated build scripts will generate the following tags for images based on

the above dockerfile paths:

```

floydhub/PROJECT_NAME:VERSION-ENV

floydhub/PROJECT_NAME:VERSION-ENV-gpu

```

Contains docker images for popular deep learning frameworks including: Tensorflow, PyTorch and Torch.