Skip to content
experimental machine learning container
Shell Dockerfile R Makefile
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Type Name Latest commit message Commit time
Failed to load latest commit information.
cuda Update test_blas.R Nov 5, 2019

Rocker stack for Machine Learning in R

This repository contains images for machine learning and GPU-based computation in R.

The dependency stack looks like so:

-| rocker/tidyverse
  -| rocker/tensorflow
    -| rocker/ml
  -| rocker/cuda 
    -| rocker/tensorflow-gpu
      -| rocker/ml-gpu
    -| rocker/cuda-dev

Nvidia CUDA libraries to the rocker-versioned stack (building on rocker/tidyverse).

Quick start

Begin with the ml or ml-gpu version for a batteries-included setup

Note: gpu images require nvidia-docker runtime to run!

Run a bash shell or R command line:

docker run --rm -ti rocker/ml R
nvidia-docker run --rm -ti rocker/ml-gpu R

Or run in RStudio instance:

docker run -e PASSWORD=mu -p 8787:8787 rocker/ml
nvidia-docker run -e PASSWORD=mu -p 8787:8787 rocker/ml-gpu


GPU-based images are built with CUDA 9.0 (on latest / 3.5.2 tags, where tags track the current R release, as in the rest of the rocker-versioned stack).

You can’t perform that action at this time.