Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
https://github.com/MachineLearningSystem/geeps
GPU-specialized parameter server for GPU machine learning.
https://github.com/MachineLearningSystem/geeps
Last synced: about 1 month ago
JSON representation
GPU-specialized parameter server for GPU machine learning.
- Host: GitHub
- URL: https://github.com/MachineLearningSystem/geeps
- Owner: MachineLearningSystem
- License: other
- Fork: true (cuihenggang/geeps)
- Created: 2022-05-11T13:12:27.000Z (over 2 years ago)
- Default Branch: master
- Last Pushed: 2018-04-05T21:15:12.000Z (over 6 years ago)
- Last Synced: 2024-08-02T19:33:33.557Z (5 months ago)
- Homepage:
- Size: 135 KB
- Stars: 0
- Watchers: 0
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
- awesome-AI-system - GeePS: Scalable Deep Learning on Distributed GPUs with a GPU-Specialized Parameter Server Eurosys'16
README
# GeePS
[](LICENSE)
[GeePS](https://cuihenggang.github.io/archive/paper/[eurosys16]geeps.pdf) is a parameter server library that scales single-machine GPU machine learning applications (such as Caffe) to a cluster of machines.
## Download and build GeePS and Caffe application
Run the following command to download GeePS and (our slightly modified) Caffe:
```
git clone --recurse-submodules https://github.com/cuihenggang/geeps.git
```
If you use the Ubuntu 14.04 system, you can run the following commands (from geeps root directory) to install the dependencies:
```
./scripts/install-geeps-deps-ubuntu14.sh
./scripts/install-caffe-deps-ubuntu14.sh
```
Also, please make sure your CUDA library is installed in `/usr/local/cuda`.
After installing the dependencies, you can build GeePS by simply running this command from geeps root directory:
```
scons -j8
```
You can then build (our slightly modified) Caffe by first entering the `apps/caffe` directory and then running `make -j8`:
```
cd apps/caffe
make -j8
```
## Caffe's CIFAR-10 example on two machines
You can run Caffe distributedly across a cluster of machines with GeePS. In this section, we will show you the steps to run Caffe's CIFAR-10 example on two machines.
All commands in this section are executed from the `apps/caffe` directory:
```
cd apps/caffe
```
You will first need to prepare a machine file as `examples/cifar10/2parts/machinefile`, with each line being the host name of one machine. Since we use two machines in this example, this machine file should have two lines, such as:
```
host0
host1
```
We will use `pdsh` to launch commands on those machines with the `ssh` protocol, so please make sure that you can `ssh` to those machines without password.
When you have your machine file in ready, you can run the following command to download and prepare the CIFAR-10 dataset:
```
./data/cifar10/get_cifar10.sh
./examples/cifar10/2parts/create_cifar10_pdsh.sh
```
Our script will partition the datasets into two parts, one for each machine. You can then train an Inception network on it with this command:
```
./examples/cifar10/2parts/train_inception.sh
```
Please look at our [wiki](https://github.com/cuihenggang/geeps/wiki) for more details. Happy training!
## Automatic training hyperparameter tuning
[MLtuner-GeePS](https://github.com/cuihenggang/mltuner-geeps) is an extended version of GeePS with automatic training hyperparameter tuning support. It includes a lightweight [MLtuner](https://cuihenggang.github.io/archive/paper/[arxiv]mltuner.pdf) module that automatically tunes the training hyperparameters for distributed ML training (including learning rate, momentum, batch size, data staleness, etc).
## Reference Paper
Henggang Cui, Hao Zhang, Gregory R. Ganger, Phillip B. Gibbons, and Eric P. Xing.
[GeePS: Scalable Deep Learning on Distributed GPUs with a GPU-Specialized Parameter Server](https://cuihenggang.github.io/archive/paper/[eurosys16]geeps.pdf).
In ACM European Conference on Computer Systems, 2016 (EuroSys'16).
Henggang Cui, Gregory R. Ganger, and Phillip B. Gibbons.
[MLtuner: System Support for Automatic Machine Learning Tuning](https://cuihenggang.github.io/archive/paper/[arxiv]mltuner.pdf).
arXiv preprint 1803.07445.