/chef-nvidia

NVIDIA Driver bootstrap for running GPU enabled applications (Caffe, Torch7, TensorFlow).

Primary LanguageRubyOtherNOASSERTION

chef-nvidia-cookbook

NVIDIA Driver bootstrap for running GPU enabled applications (Caffe, Torch7, TensorFlow).

Supported Platforms

Written with Amazon Linux AMI in mind, but should work on Ubuntu, etc.

Attributes

Key Type Description Default
[:nvidia][:arch] String the architecture for the driver x86_64
[:nvidia][:version] String the version for the driver 364.19

Usage

chef-nvidia::default

Include chef-nvidia in your node's run_list:

{
  "run_list": [
    "recipe[chef-nvidia::default]"
  ]
}

License and Authors

Author:: Brad Folkens (bfolkens@gmail.com)