rsnk96 / Ubuntu-Setup-Scripts

Scripts to help you set up your Ubuntu quickly, especially if you're in any subfield of Data Science or AI!
Apache License 2.0
128 stars 76 forks source link

Add nvidia-docker2 installation #46

Closed rajat2004 closed 5 years ago

rajat2004 commented 5 years ago

It has been commented out since this should be installed only if an Nvidia GPU is present Things to note-

  1. nvidia-docker should be uninstalled before installing nvidia-docker2. This can be added.
  2. We can check for the presence of the GPU using nvidia-smi and then only install

Source - https://github.com/NVIDIA/nvidia-docker#quickstart

rajat2004 commented 5 years ago

14 can be closed after this goes in

Should we have nvidia-docker installation also?

rajat2004 commented 5 years ago

@rsnk96 Changes made

rajat2004 commented 5 years ago

Added missing -y flag, good to go now