Skip to content
Branch: master
Find file History
Type Name Latest commit message Commit time
Failed to load latest commit information. updated docker folder (#2316) Aug 13, 2019
dockerfile.gpu updated dockers (#2459) Oct 7, 2019

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-cudnn5-devel image. LightGBM can be utilized in GPU and CPU modes and via Python (2.7 & 3.6).


  • LightGBM (cpu + gpu)
  • Python 2.7 (conda) + scikit-learn, notebooks, pandas, matplotlib
  • Python 3.6 (conda) + scikit-learn, notebooks, pandas, matplotlib

Running the container starts a Jupyter Notebook at localhost:8888.

Jupyter password: keras.


Requires docker and nvidia-docker on host machine.


Build Docker Image

mkdir lightgbm-docker
cd lightgbm-docker
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

You can’t perform that action at this time.