Public | Automated Build

Last pushed: 6 months ago
Short Description
Docker image for machine learning on GPU
Full Description

Docker image for machine learning on GPU

Installed packages

  • scikit-learn
  • numpy
  • scipy
  • pandas
  • awscli
  • click
  • tqdm
  • torch

Usage example on AWS

First, create GPU instance on AWS and ssh to the remote machine.
See official wiki: Deploy on Amazon EC2.

Then, on the remote machine try

$ nvidia-docker run --rm -it sotetsuk/nvidia-ml python
Python 3.6.0 |Continuum Analytics, Inc.| (default, Dec 23 2016, 12:22:00)
[GCC 4.4.7 20120313 (Red Hat 4.4.7-1)] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import numpy as np
Docker Pull Command
Owner
sotetsuk
Source Repository