Skip to content

Commit

Permalink
add qgis,ros and badge
Browse files Browse the repository at this point in the history
  • Loading branch information
mhwasil committed Apr 25, 2024
1 parent 0f56957 commit e4e44a9
Showing 1 changed file with 17 additions and 10 deletions.
27 changes: 17 additions & 10 deletions README.md
Original file line number Diff line number Diff line change
@@ -1,29 +1,36 @@
# A2S Institute Docker Images

[![Build, test, and push images](https://github.com/a2s-institute/docker-stacks/actions/workflows/docker.yml/badge.svg)](https://github.com/a2s-institute/docker-stacks/actions/workflows/docker.yml)

Our stacks provide GPU-enabled Jupyter Notebook in Docker containers, which can also run on Kubernetes. The images are based on [Jupyter docker-stacks jupyter/pytorch-notebook](https://github.com/jupyter/docker-stacks/tree/main/images/pytorch-notebook). All images are published on our [ghcr.io](https://github.com/orgs/a2s-institute/packages) and [quay.io](https://quay.io/user/a2s-institute/).

The stacks contain several machine learning packages such as TensorFlow, PyTorch, scikit-learn, and other machine learning tools. All images also include VSCode and xfce4 desktop environment.

## Docker stack structure
* [gpu-base-notebook](https://github.com/a2s-institute/docker-stacks/tree/master/base-gpu-notebook): contains Jupyter related libraries and also includes different cuda and pytorch versions. It also has VSCode and xfce4 desktop environment.
* [ml-notebook](https://github.com/a2s-institute/docker-stacks/tree/master/ml-notebook): depends on `gpu-base-notebook` and includes several machine learning libaries such as TensorfLow, Keras, scipy, opencv, etc.
* [nlp-notebook](https://github.com/a2s-institute/docker-stacks/tree/master/nlp-notebook): depends on `ml-notebook` and includes NLP libraries such as spaCy, NLTK, llama-cpp-python and wikipedia-api.
* [gpu-base-notebook:cuda*-pytorch*](https://github.com/a2s-institute/docker-stacks/tree/master/base-gpu-notebook): contains Jupyter related libraries and also includes different cuda and pytorch versions. It also has VSCode and xfce4 desktop environment.
* [ml-notebook:cuda*-pytorch*](https://github.com/a2s-institute/docker-stacks/tree/master/ml-notebook): depends on `gpu-base-notebook` and includes several machine learning libaries such as TensorfLow, Keras, scipy, opencv, etc.
* [nlp-notebook:cuda*-pytorch*](https://github.com/a2s-institute/docker-stacks/tree/master/nlp-notebook): depends on `ml-notebook` and includes NLP libraries such as spaCy, NLTK, llama-cpp-python and wikipedia-api.
* [qgis:cuda*-v*](https://github.com/a2s-institute/docker-stacks/tree/master/qgis/cuda12-v3.36.1): depends on `gpu-base-notebook`, but does not have all the ML related libraries as in `ml-notebook`.
* [gpu-base-notebook:cuda12-ubuntu12.04](https://github.com/a2s-institute/docker-stacks/tree/master/base-gpu-notebook/cuda12-ubuntu20.04): based on `nvidia/cuda:12.3.2-cudnn9-runtime-ubuntu20.04` and has Jupyter Notebook related components with Ubuntu 20.04.
* [ros:cuda12-noetic](https://github.com/a2s-institute/docker-stacks/tree/master/ros/cuda12-noetic): based on `gpu-base-notebook:cuda12-ubuntu20.04` and includes `ros-noetic-desktop-full` with the gazebo simulation, but does not have all the ML related libraries as in `ml-notebook`.


## Avilable versions
* `gpu-base-notebook:cuda11-pytorch-2.2.2`
* `gpu-base-notebook:cuda12-pytorch-2.2.2`
* `ml-notebook:cuda11-pytorch-2.2.2`
* `ml-notebook:cuda12-pytorch-2.2.2`
* `nlp-notebook:cuda11-pytorch-2.2.2`
* `nlp-notebook:cuda12-pytorch-2.2.2`
* `quay.io/a2s-institute/gpu-base-notebook:cuda12-pytorch-2.2.2`
* `quay.io/a2s-institute/ml-notebook:cuda12-pytorch-2.2.2`
* `quay.io/a2s-institute/nlp-notebook:cuda12-pytorch-2.2.2`
* `quay.io/a2s-institute/qgis:cuda12-v3.36.1`
* `quay.io/a2s-institute/ros:cuda12-noetic`
* `quay.io/a2s-institute/gpu-base-notebook:cuda11-pytorch-2.2.2`
* `quay.io/a2s-institute/ml-notebook:cuda11-pytorch-2.2.2`
* `quay.io/a2s-institute/nlp-notebook:cuda11-pytorch-2.2.2`

<details>
<summary><font color=blue> Older images</font></summary>

- `ghcr.io/a2s-institute/docker-stacks/gpu-notebook:cuda11.3.1-ubuntu22.04` (no vscode and xfce desktop)
- `ghcr.io/a2s-institute/docker-stacks/gpu-notebook:cuda11.8.0-ubuntu22.04` (no vscode and xfce desktop)
- `ghcr.io/a2s-institute/docker-stacks/gpu-notebook:cuda12.1.0-ubuntu22.04` (no vscode and xfce desktop)
- `ghcr.io/a2s-institute/docker-stacks/gpu-notebook:cuda12.1.0-ubuntu22.04` (no vscode and xfce desktop)

</details>

Expand Down

0 comments on commit e4e44a9

Please sign in to comment.