README.md 1.13 KB
Newer Older
1
2
# Dockerfile for LightGBM GPU Version with Python

Nikita Titov's avatar
Nikita Titov committed
3
4
A docker file with LightGBM utilizing nvidia-docker. The file is based on the `nvidia/cuda:8.0-cudnn5-devel` image.
LightGBM can be utilized in GPU and CPU modes and via Python (2.7 & 3.6).
5
6
7

## Contents

8
- LightGBM (cpu + gpu)
Nikita Titov's avatar
Nikita Titov committed
9
10
- Python 2.7 (conda) + scikit-learn, notebooks, pandas, matplotlib
- Python 3.6 (conda) + scikit-learn, notebooks, pandas, matplotlib
11

Nikita Titov's avatar
Nikita Titov committed
12
Running the container starts a Jupyter Notebook at `localhost:8888`.
13

Nikita Titov's avatar
Nikita Titov committed
14
Jupyter password: `keras`.
15
16
17

## Requirements

18
19
Requires docker and [nvidia-docker](https://github.com/NVIDIA/nvidia-docker) on host machine.

20
21
22
23
## Quickstart

### Build Docker Image

24
25
26
```sh
mkdir lightgbm-docker
cd lightgbm-docker
firasrb's avatar
firasrb committed
27
wget https://raw.githubusercontent.com/Microsoft/LightGBM/master/docker/gpu/dockerfile.gpu
28
29
docker build -f dockerfile.gpu -t lightgbm-gpu .
```
30
31
32

### Run Image

33
34
35
36
```sh
nvidia-docker run --rm -d --name lightgbm-gpu -p 8888:8888 -v /home:/home lightgbm-gpu
```

37
38
### Attach with Command Line Access (if required)

39
40
41
```sh
docker exec -it lightgbm-gpu bash
```
42
43
44

### Jupyter Notebook

45
46
47
```sh
localhost:8888
```