Sign inSign up

sotetsuk/nvidia-ml

By sotetsuk

•Updated over 9 years ago

Docker image for machine learning on GPU

Image
0

396

sotetsuk/nvidia-ml repository overview

⁠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

Tag summary

Content type

Image

Digest

Size

2.6 GB

Last updated

over 9 years ago

docker pull sotetsuk/nvidia-ml