Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
https://github.com/ankane/ml-stack
Machine learning Docker images for Ruby
https://github.com/ankane/ml-stack
Last synced: 12 days ago
JSON representation
Machine learning Docker images for Ruby
- Host: GitHub
- URL: https://github.com/ankane/ml-stack
- Owner: ankane
- License: mit
- Created: 2020-05-01T07:00:45.000Z (over 4 years ago)
- Default Branch: master
- Last Pushed: 2023-09-09T21:54:43.000Z (about 1 year ago)
- Last Synced: 2024-10-03T12:40:36.452Z (about 1 month ago)
- Language: Jupyter Notebook
- Size: 32.2 KB
- Stars: 32
- Watchers: 3
- Forks: 3
- Open Issues: 1
-
Metadata Files:
- Readme: README.md
- License: LICENSE.txt
Awesome Lists containing this project
README
# ml-stack
:fire: Machine learning Docker images for Ruby
Spin up a machine learning environment in a single command
[](https://mybinder.org/v2/gh/ankane/ml-stack/master)
## Getting Started
- [Docker](#docker)
- [Paperspace](#paperspace)
- [Binder](#binder)
### Docker
Start a container
```sh
docker run -p 8888:8888 ankane/ml-stack:standard
```
And visit `http://127.0.0.1:8888/?token=...`.
Share a volume with the container
```sh
-v /path/on/host:/notebooks
```
For GPU images, install the NVIDIA driver and [container toolkit](https://github.com/NVIDIA/nvidia-docker) on the host machine. Then use:
```sh
--gpus all
```
### Paperspace
Use a [custom container](https://docs.paperspace.com/gradient/notebooks/notebook-containers) for your notebook. Set the container name to the image you want to use, like:
```text
ankane/ml-stack:torch-gpu
```
And leave the other fields in that section blank.
### Binder
Create a `Dockerfile` in the root of your repository with the image you want to use and your notebooks:
```Dockerfile
# Choose an image
FROM ankane/ml-stack:standard
# Remove example notebooks
RUN rm *.ipynb
# Copy your notebooks
COPY Notebook1.ipynb Notebook2.ipynb ./
# The rest is specific to Binder
ARG NB_USER
ARG NB_UID
RUN adduser --disabled-password --gecos '' --uid ${NB_UID} ${NB_USER} && \
chown -R ${NB_USER}:${NB_USER} .
USER ${NB_USER}
RUN mkdir ~/.jupyter && \
echo 'c.KernelSpecManager.ensure_native_kernel = False' > ~/.jupyter/jupyter_notebook_config.py && \
iruby register
```
Then visit [Binder](https://mybinder.org) to run it.
## Images
There are a few images to choose from:
- [Standard](#standard)
- [Torch GPU](#torch-gpu)
### Standard
Tag: `standard`
- [Daru](https://github.com/SciRuby/daru)
- [fastText](https://github.com/ankane/fastText-ruby)
- [Informers](https://github.com/ankane/informers)
- [LIBMF](https://github.com/ankane/libmf-ruby)
- [LightGBM](https://github.com/ankane/lightgbm-ruby)
- [Numo](https://github.com/ruby-numo/numo-narray)
- [Nyaplot](https://github.com/domitry/nyaplot)
- [Polars](https://github.com/ankane/ruby-polars)
- [Prophet](https://github.com/ankane/prophet-ruby)
- [Rover](https://github.com/ankane/rover)
- [Rumale](https://github.com/yoshoku/rumale)
- [Vega](https://github.com/ankane/vega-ruby)
- [XGBoost](https://github.com/ankane/xgboost-ruby)
- [xLearn](https://github.com/ankane/xlearn-ruby)
### Torch GPU
Tag: `torch-gpu`
Designed for deep learning with Torch
- [Torch.rb](https://github.com/ankane/torch.rb)
- [TorchAudio](https://github.com/ankane/torchaudio-ruby)
- [TorchRec](https://github.com/ankane/torchrec-ruby)
- [TorchText](https://github.com/ankane/torchtext-ruby)
- [TorchVision](https://github.com/ankane/torchvision-ruby)
## Contributing
Everyone is encouraged to help improve this project. Here are a few ways you can help:
- [Report bugs](https://github.com/ankane/ml-stack/issues)
- Fix bugs and [submit pull requests](https://github.com/ankane/ml-stack/pulls)
- Write, clarify, or fix documentation
- Suggest or add new features
To get started with development:
```sh
git clone https://github.com/ankane/ml-stack.git
cd ml-stack
# build specific images
docker build -f standard/Dockerfile -t ankane/ml-stack:standard .
docker build -f torch-gpu/Dockerfile -t ankane/ml-stack:torch-gpu .
```