Permalink
Type Name Latest commit message Commit time
..
Failed to load latest commit information.
README.md Updated wget for GPU docker (#1404) May 30, 2018
dockerfile.gpu gpu enabled docker file with python bindings (#929) Sep 29, 2017

README.md

Dockerfile for LightGBM GPU Version with Python

A docker file with LightGBM utilizing nvidia-docker. The file is based on the nvidia/cuda:8.0 image. LightGBM can be utilized in GPU and CPU modes and via Python (2.7 & 3.5)

Contents

  • LightGBM (cpu + gpu)
  • Python 2.7 (Conda) + scikit-learn notebooks pandas matplotlib
  • Python 3.5 (Conda) + scikit-learn notebooks pandas matplotlib

Running the container starts a jupyter notebook at localhost:8888

jupyter password: keras

Requirements

Requires docker and nvidia-docker on host machine.

Quickstart

Build Docker Image

mkdir lightgbm-docker
cd lightgbm-docker
wget https://raw.githubusercontent.com/Microsoft/LightGBM/master/docker/gpu/dockerfile.gpu
docker build -f dockerfile.gpu -t lightgbm-gpu .

Run Image

nvidia-docker run --rm -d --name lightgbm-gpu -p 8888:8888 -v /home:/home lightgbm-gpu

Attach with Command Line Access (if required)

docker exec -it lightgbm-gpu bash

Jupyter Notebook

localhost:8888