NVIDIA Driver bootstrap for running GPU enabled applications (Caffe, Torch7, TensorFlow).
Written with Amazon Linux AMI in mind, but should work on Ubuntu, etc.
Key | Type | Description | Default |
---|---|---|---|
[:nvidia][:arch] | String | the architecture for the driver | x86_64 |
[:nvidia][:version] | String | the version for the driver | 364.19 |
Include chef-nvidia
in your node's run_list
:
{
"run_list": [
"recipe[chef-nvidia::default]"
]
}
Author:: Brad Folkens ([email protected])